refac: fix custom_lists
This commit is contained in:
parent
474588678a
commit
16c64f668a
2 changed files with 39 additions and 2 deletions
|
|
@ -1,6 +1,43 @@
|
||||||
|
## LLM
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
from langchain import llms
|
from langchain import llms
|
||||||
from langchain.llms.openai import OpenAIChat
|
from langchain.llms.openai import OpenAIChat
|
||||||
|
|
||||||
|
|
||||||
llm_type_to_cls_dict = llms.type_to_cls_dict
|
llm_type_to_cls_dict = llms.type_to_cls_dict
|
||||||
llm_type_to_cls_dict["openai-chat"] = OpenAIChat
|
llm_type_to_cls_dict["openai-chat"] = OpenAIChat
|
||||||
|
|
||||||
|
|
||||||
|
## Memory
|
||||||
|
|
||||||
|
# from langchain.memory.buffer_window import ConversationBufferWindowMemory
|
||||||
|
# from langchain.memory.chat_memory import ChatMessageHistory
|
||||||
|
# from langchain.memory.combined import CombinedMemory
|
||||||
|
# from langchain.memory.entity import ConversationEntityMemory
|
||||||
|
# from langchain.memory.kg import ConversationKGMemory
|
||||||
|
# from langchain.memory.readonly import ReadOnlySharedMemory
|
||||||
|
# from langchain.memory.simple import SimpleMemory
|
||||||
|
# from langchain.memory.summary import ConversationSummaryMemory
|
||||||
|
# from langchain.memory.summary_buffer import ConversationSummaryBufferMemory
|
||||||
|
|
||||||
|
memory_type_to_cls_dict: dict[str, Any] = {
|
||||||
|
# "CombinedMemory": CombinedMemory,
|
||||||
|
# "ConversationBufferWindowMemory": ConversationBufferWindowMemory,
|
||||||
|
# "ConversationBufferMemory": ConversationBufferMemory,
|
||||||
|
# "SimpleMemory": SimpleMemory,
|
||||||
|
# "ConversationSummaryBufferMemory": ConversationSummaryBufferMemory,
|
||||||
|
# "ConversationKGMemory": ConversationKGMemory,
|
||||||
|
# "ConversationEntityMemory": ConversationEntityMemory,
|
||||||
|
# "ConversationSummaryMemory": ConversationSummaryMemory,
|
||||||
|
# "ChatMessageHistory": ChatMessageHistory,
|
||||||
|
# "ConversationStringBufferMemory": ConversationStringBufferMemory,
|
||||||
|
# "ReadOnlySharedMemory": ReadOnlySharedMemory,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
## Chain
|
||||||
|
# from langchain.chains.loading import type_to_loader_dict
|
||||||
|
# from langchain.chains.conversation.base import ConversationChain
|
||||||
|
|
||||||
|
# chain_type_to_cls_dict = type_to_loader_dict
|
||||||
|
# chain_type_to_cls_dict["conversation_chain"] = ConversationChain
|
||||||
|
|
|
||||||
|
|
@ -54,5 +54,5 @@
|
||||||
"last 1 safari version"
|
"last 1 safari version"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
"proxy": "http://backend:7860"
|
"proxy": "http://localhost:7860"
|
||||||
}
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue