shaytan/ai-agents-masterclass
1
0
Fork 0
mirror of https://github.com/coleam00/ai-agents-masterclass.git synced 2025-11-29 16:43:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Couple small code changes for LangServe walkthrough

Browse Source
This commit is contained in:
Cole Medin
2024-08-30 12:41:06 -05:00
parent af684be17e
commit b3438810d7
4 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
10-deploy-ai-agent-langserve/langserve-chatbot.py
View File

@@ -9,14 +9,12 @@ import os
from langchain_core.messages import SystemMessage, AIMessage, HumanMessage, ToolMessage
from langserve import RemoteRunnable
from runnable import get_runnable
load_dotenv()
agent_endpoint_url = os.getenv('AGENT_ENDPOINT_URL', 'http://localhost:8000')
@st.cache_resource
def create_chatbot_instance():
return RemoteRunnable(f"{agent_endpoint_url}/invoke/")
return RemoteRunnable(agent_endpoint_url)
chatbot = create_chatbot_instance()