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

phidata -> agno

Browse source
This commit is contained in:
Madhu 2025-03-17 20:16:52 +05:30
parent fc02a7f3dd
commit 93176697d1
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
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
@ -75,10 +75,9 @@ def initialize_models():
os.environ["OPENAI_API_KEY"] = st.session_state.openai_api_key os.environ["OPENAI_API_KEY"] = st.session_state.openai_api_key
st.session_state.embeddings = OpenAIEmbeddings(model="text-embedding-3-small") 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: 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)