Gemini changes

one_fm/docsagent/config.yaml

@@ -0,0 +1,40 @@
+#
+# Copyright 2023 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+configs:
+  - product_name: "WIKI"
+    models:
+      - language_model: "models/gemini-1.5-flash-latest"
+        embedding_model: "models/embedding-001"
+        api_endpoint: "generativelanguage.googleapis.com"
+        embedding_api_call_limit: 1400
+        embedding_api_call_period: 60
+    docs_agent_config: "normal"
+    markdown_splitter: "token_splitter"
+    log_level: "NORMAL"
+    db_type: "chroma"
+    db_configs:
+      - db_type: "chroma"
+        vector_db_dir: "vector_stores/chroma"
+        collection_name: "docs_collection"
+    output_path: "data/plain_docs"
+    inputs:
+      - path: "content" # create a content folder inside docsagent dir give the absolute path  
+        url_prefix: "http://onefm.localhost:8003/app/wiki"
+    conditions:
+      - condition_text: "You are a helpful chatbot answering questions from users. Read the context below first and answer the user's question at the end. In your answer, provide a summary in three or five sentences. (BUT DO NOT USE ANY INFORMATION YOU KNOW ABOUT THE WORLD.)"
+        fact_check_question: "Can you compare the text below to the information provided in this prompt above and write a short message that warns the readers about which part of the text they should consider fact-checking? (Please keep your response concise, focus on only one important item, but DO NOT USE BOLD TEXT IN YOUR RESPONSE.)"
+        model_error_message: "Gemini is not able to answer this question at the moment. Rephrase the question and try asking again."