From 93176697d10bfb3509a3226897423ecfcd8a56e8 Mon Sep 17 00:00:00 2001 From: Madhu Date: Mon, 17 Mar 2025 20:16:52 +0530 Subject: [PATCH 1/2] phidata -> agno --- .../rag_database_routing/rag_database_routing.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/rag_tutorials/rag_database_routing/rag_database_routing.py b/rag_tutorials/rag_database_routing/rag_database_routing.py index e008121..c088671 100644 --- a/rag_tutorials/rag_database_routing/rag_database_routing.py +++ b/rag_tutorials/rag_database_routing/rag_database_routing.py @@ -9,8 +9,8 @@ from langchain_community.vectorstores import Qdrant from langchain_openai import OpenAIEmbeddings from langchain_openai import ChatOpenAI import tempfile -from phi.agent import Agent -from phi.model.openai import OpenAIChat +from agno.agent import Agent +from agno.models.openai import OpenAIChat from langchain.schema import HumanMessage from langchain.chains.combine_documents import create_stuff_documents_chain from langchain.chains import create_retrieval_chain @@ -75,10 +75,9 @@ def initialize_models(): os.environ["OPENAI_API_KEY"] = st.session_state.openai_api_key st.session_state.embeddings = OpenAIEmbeddings(model="text-embedding-3-small") - st.session_state.llm = ChatOpenAI(temperature=0) + st.session_state.model = ChatOpenAI(temperature=0) try: - # Initialize Qdrant client with session state credentials client = QdrantClient( url=st.session_state.qdrant_url, api_key=st.session_state.qdrant_api_key @@ -259,7 +258,7 @@ def query_database(db: Qdrant, question: str) -> tuple[str, list]: except Exception as e: st.error(f"Error: {str(e)}") return "I encountered an error. Please try rephrasing your question.", [] - + def _handle_web_fallback(question: str) -> tuple[str, list]: st.info("No relevant documents found. Searching web...") fallback_agent = create_fallback_agent(st.session_state.llm) From 9a812ee04ca8ffdbc47c1aebd5ae8a86ff12859a Mon Sep 17 00:00:00 2001 From: Madhu Date: Mon, 17 Mar 2025 20:18:43 +0530 Subject: [PATCH 2/2] final changes --- rag_tutorials/rag_database_routing/rag_database_routing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rag_tutorials/rag_database_routing/rag_database_routing.py b/rag_tutorials/rag_database_routing/rag_database_routing.py index c088671..7e83a68 100644 --- a/rag_tutorials/rag_database_routing/rag_database_routing.py +++ b/rag_tutorials/rag_database_routing/rag_database_routing.py @@ -75,7 +75,7 @@ def initialize_models(): os.environ["OPENAI_API_KEY"] = st.session_state.openai_api_key st.session_state.embeddings = OpenAIEmbeddings(model="text-embedding-3-small") - st.session_state.model = ChatOpenAI(temperature=0) + st.session_state.llm = ChatOpenAI(temperature=0) try: client = QdrantClient(