Merge pull request #80 from h2oai/update_company_overview

Update company overview to reflect h2ogpte 1.5.10

company-financial-overview/app.toml

@@ -1,6 +1,6 @@
 [App]
 Name = "ai.h2o.wave.company_info_at_a_glance"
-Version = "0.0.6"
+Version = "0.0.7"
 Title = "Company Information"
 Description = "Use AI to research companies and identify opportunities"
 Tags = ["GENERATIVE_AI", "DATA_SCIENCE", "OSS"]

company-financial-overview/prompts/competitor_prompt.txt

@@ -1 +1 @@
-Who are the main competitors for the company provided? Can you list them by market cap and number of employees?
+Who are the main competitors for the company provided? Please list them by market cap and number of employees?

company-financial-overview/requirements.txt

@@ -1,4 +1,4 @@
-h2ogpte==1.4.9
+h2ogpte==1.5.10
 h2o_wave
 loguru
 toml

company-financial-overview/src/generate_content.py

@@ -125,7 +125,7 @@ def stream_response(message):
         client = H2OGPTE(address=os.getenv("H2OGPTE_URL"), api_key=os.getenv("H2OGPTE_API_TOKEN"))
 
         collection_id = client.create_collection("Temp for Company Financial Overview", "")
-        chat_session_id = client.create_chat_session(collection_id)
+        chat_session_id = client.create_chat_session(collection_id=collection_id)
         print("USER MESSSAGEEEEE", chatbot_interaction.user_message)
         with client.connect(chat_session_id) as session:
             session.query(
@@ -159,6 +159,6 @@ def update_response(self, message):
             self.content_to_show = message.content
             self.responding = False
         elif isinstance(message, PartialChatMessage):
-            if message.content != "#### LLM Only (no RAG):\n":
+            if message.content != "#### LLM Only:\n":
                 self.llm_response += message.content
                 self.content_to_show = self.llm_response + " 🟡"