admin-valentin/awesome-llm-apps
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #149 from Madhuvod/database-routing

Browse source 
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
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
rag_tutorials/rag_database_routing/rag_database_routing.py
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
@ -259,7 +258,7 @@ def query_database(db: Qdrant, question: str) -> tuple[str, list]:
except Exception as e: except Exception as e:
st.error(f"Error: {str(e)}") st.error(f"Error: {str(e)}")
return "I encountered an error. Please try rephrasing your question.", [] return "I encountered an error. Please try rephrasing your question.", []
def _handle_web_fallback(question: str) -> tuple[str, list]: def _handle_web_fallback(question: str) -> tuple[str, list]:
st.info("No relevant documents found. Searching web...") st.info("No relevant documents found. Searching web...")
fallback_agent = create_fallback_agent(st.session_state.llm) fallback_agent = create_fallback_agent(st.session_state.llm)