shaytan/ai-agents-masterclass
1
0
Fork 0
mirror of https://github.com/coleam00/ai-agents-masterclass.git synced 2025-12-01 09:33:12 +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

10
10-deploy-ai-agent-langserve/runnable.py
View File

@@ -14,8 +14,8 @@ from langchain_anthropic import ChatAnthropic
from langchain_core.messages import ToolMessage, AIMessage
from tools.asana_tools import available_asana_functions
from tools.google_drive_tools import available_drive_functions
from tools.vector_db_tools import available_vector_db_functions
# from tools.google_drive_tools import available_drive_functions
# from tools.vector_db_tools import available_vector_db_functions
load_dotenv()
model = os.getenv('LLM_MODEL', 'gpt-4o')
@@ -26,7 +26,11 @@ model_mapping = {
"groq": ChatGroq
}
available_functions = available_asana_functions | available_drive_functions | available_vector_db_functions
# Google Drive and Vector DB tools are not included by default because you'll need a bigger cloud instance (DigitalOcean droplet)
# To load the Vector DB and add Google Drive documents to the knowledgebase. You could also use something else like
# Pinecone instead of running Chroma locally! Uncomment lines 17 and 18 + replace line 33 with 32 if you want to use these tools.
# available_functions = available_asana_functions | available_drive_functions | available_vector_db_functions
available_functions = available_asana_functions
tools = [tool for _, tool in available_functions.items()]
for key, chatbot_class in model_mapping.items():