fix bug related to OpenAI endpoint

app.py

@@ -5,8 +5,8 @@ from pinecone import Pinecone
 from huggingface_hub import whoami
 from langchain.prompts import ChatPromptTemplate
 from langchain_community.embeddings import HuggingFaceBgeEmbeddings
-from langchain_openai import AzureChatOpenAI
 from langchain.prompts.prompt import PromptTemplate
+from langchain_groq import ChatGroq
 from langchain.memory import ConversationBufferMemory
 from langchain_community.vectorstores import Pinecone as PineconeVectorstore
 from celsius_csrd_chatbot.utils import (
@@ -14,14 +14,12 @@ from celsius_csrd_chatbot.utils import (
     make_pairs,
     _format_chat_history,
     _combine_documents,
-    get_llm,
     init_env,
     parse_output_llm_with_sources,
 )
 from celsius_csrd_chatbot.agent import make_graph_agent, display_graph
 
 init_env()
-chat_model_init = get_llm()
 demo_name = "ESRS_QA"
 hf_model = "BAAI/bge-base-en-v1.5"
 
@@ -33,7 +31,7 @@ embeddings = HuggingFaceBgeEmbeddings(
 pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
 index = pc.Index(os.getenv("PINECONE_API_INDEX"))
 vectorstore = PineconeVectorstore(index, embeddings, "page_content")
-llm = AzureChatOpenAI()
+llm = ChatGroq(temperature=0, model_name="llama-3.2-90b-text-preview")
 agent = make_graph_agent(llm, vectorstore)
 
 memory = ConversationBufferMemory(

celsius_csrd_chatbot/chains/esrs_categorization.py

@@ -5,7 +5,7 @@ def make_esrs_categorization_node():
 
     def categorize_message(state):
         query = state["query"]
-        pattern = r"ESRS \d+[A-Z0-9]*"
+        pattern = r"ESRS \d+[A-Z0-9]*|ESRS [A-Z]+\d+[A-Z0-9]*"
         esrs_truth = [
             "ESRS 1",
             "ESRS 2",

celsius_csrd_chatbot/utils.py

@@ -3,7 +3,6 @@ import re
 from typing import Tuple, List
 from dotenv import load_dotenv
 from msal import ConfidentialClientApplication
-from langchain_openai import AzureChatOpenAI
 from langchain.schema import format_document
 
 

pyproject.toml

@@ -8,19 +8,18 @@ package-mode = true
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
-langchain = "^0.2.5"
 gradio = {extras = ["oauth"], version = "^4.36.1"}
 sentence-transformers = "^3.0.1"
-langchain-community = "^0.2.5"
 msal = "^1.28.1"
 loadenv = "^0.1.1"
 openai = "^1.34.0"
-langchain-openai = "^0.1.8"
 pinecone = "^4.0.0"
 pinecone-client = "^5.0.1"
 langgraph = "^0.2.0"
-langchain-core = "^0.2.29"
 ipython = "^8.26.0"
+langchain-groq = "^0.2.0"
+langchain-core = "^0.3.12"
+langchain = "^0.3.3"
 
 
 [build-system]