phidata -> agno
This commit is contained in:
parent
fc02a7f3dd
commit
93176697d1
1 changed files with 4 additions and 5 deletions
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue