diff --git a/company-financial-overview/app.toml b/company-financial-overview/app.toml index 39cc6ca..0dd16b4 100644 --- a/company-financial-overview/app.toml +++ b/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"] diff --git a/company-financial-overview/prompts/competitor_prompt.txt b/company-financial-overview/prompts/competitor_prompt.txt index dcfd278..487a52e 100644 --- a/company-financial-overview/prompts/competitor_prompt.txt +++ b/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? \ No newline at end of file +Who are the main competitors for the company provided? Please list them by market cap and number of employees? \ No newline at end of file diff --git a/company-financial-overview/requirements.txt b/company-financial-overview/requirements.txt index e500842..fab4bfc 100644 --- a/company-financial-overview/requirements.txt +++ b/company-financial-overview/requirements.txt @@ -1,4 +1,4 @@ -h2ogpte==1.4.9 +h2ogpte==1.5.10 h2o_wave loguru toml diff --git a/company-financial-overview/src/generate_content.py b/company-financial-overview/src/generate_content.py index 1bdb0f5..9d5e63e 100644 --- a/company-financial-overview/src/generate_content.py +++ b/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 + " 🟡"