rzmk/learnhouse
1
0
Fork 0
mirror of https://github.com/rzmk/learnhouse.git synced 2025-12-19 04:19:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Migrated Langchain

Browse source
This commit is contained in:
Prateek Khandelwal 2024-10-27 16:49:22 +05:30
parent 93c41d02b4
commit 6ad20bddad
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
apps/api/src/services/ai/base.py
View file

@ -1,17 +1,17 @@
from typing import Optional from typing import Optional
from uuid import uuid4 from uuid import uuid4
from langchain.agents import AgentExecutor from langchain.agents import AgentExecutor
from langchain.text_splitter import CharacterTextSplitter from langchain_text_splitters import CharacterTextSplitter
from langchain_community.vectorstores import Chroma from langchain_community.vectorstores import Chroma
from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent
from langchain.prompts import MessagesPlaceholder from langchain_core.prompts import MessagesPlaceholder
from langchain_community.chat_message_histories import RedisChatMessageHistory from langchain_community.chat_message_histories import RedisChatMessageHistory
from langchain_core.messages import SystemMessage from langchain_core.messages import SystemMessage
from langchain.agents.openai_functions_agent.agent_token_buffer_memory import ( from langchain.agents.openai_functions_agent.agent_token_buffer_memory import (
AgentTokenBufferMemory, AgentTokenBufferMemory,
) )
from langchain_openai import OpenAIEmbeddings from langchain_openai import OpenAIEmbeddings
from langchain_community.chat_models import ChatOpenAI from langchain_openai import ChatOpenAI
from langchain.agents.agent_toolkits import ( from langchain.agents.agent_toolkits import (
create_retriever_tool, create_retriever_tool,
) )
@ -121,3 +121,6 @@ def get_chat_session_history(aichat_uuid: Optional[str] = None):
message_history = [] message_history = []
return {"message_history": message_history, "aichat_uuid": session_id} return {"message_history": message_history, "aichat_uuid": session_id}
if __name__ == '__main__':
pass