Merge pull request #149 from Madhuvod/database-routing

fix: changed from phidata to agno
This commit is contained in:
Shubham Saboo 2025-03-17 21:21:26 -05:00 committed by GitHub
commit 2d86560079
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -9,8 +9,8 @@ from langchain_community.vectorstores import Qdrant
from langchain_openai import OpenAIEmbeddings from langchain_openai import OpenAIEmbeddings
from langchain_openai import ChatOpenAI from langchain_openai import ChatOpenAI
import tempfile import tempfile
from phi.agent import Agent from agno.agent import Agent
from phi.model.openai import OpenAIChat from agno.models.openai import OpenAIChat
from langchain.schema import HumanMessage from langchain.schema import HumanMessage
from langchain.chains.combine_documents import create_stuff_documents_chain from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain.chains import create_retrieval_chain from langchain.chains import create_retrieval_chain
@ -78,7 +78,6 @@ def initialize_models():
st.session_state.llm = ChatOpenAI(temperature=0) st.session_state.llm = ChatOpenAI(temperature=0)
try: try:
# Initialize Qdrant client with session state credentials
client = QdrantClient( client = QdrantClient(
url=st.session_state.qdrant_url, url=st.session_state.qdrant_url,
api_key=st.session_state.qdrant_api_key api_key=st.session_state.qdrant_api_key