エラー修正
parent
9db4b1477b
commit
37927b9fb9
|
|
@ -5,7 +5,7 @@ import os
|
|||
import sys
|
||||
import json
|
||||
from langchain_community.llms import GPT4All, LlamaCpp
|
||||
from langchain.memory import ConversationBufferMemory
|
||||
from langchain.memory import ConversationBufferMemory, ChatMessageHistory
|
||||
from langchain.schema import (
|
||||
AIMessage,
|
||||
HumanMessage,
|
||||
|
|
@ -42,7 +42,13 @@ class LangChainApi:
|
|||
|
||||
def __init__(self, **kwargs):
|
||||
self.backend = None
|
||||
self.memory = None
|
||||
|
||||
self.memory = ConversationBufferMemory(
|
||||
human_prefix="Human",
|
||||
ai_prefix="AI",
|
||||
memory_key="history",
|
||||
return_messages=True,
|
||||
)
|
||||
|
||||
self.load_settings(**kwargs)
|
||||
|
||||
|
|
@ -72,13 +78,6 @@ class LangChainApi:
|
|||
)
|
||||
is_chat = False
|
||||
|
||||
self.memory = ConversationBufferMemory(
|
||||
human_prefix="Human",
|
||||
ai_prefix="AI",
|
||||
memory_key="history",
|
||||
return_messages=True,
|
||||
)
|
||||
|
||||
self.pydantic_parser = PydanticOutputParser(pydantic_object=Txt2ImgModel)
|
||||
|
||||
if not is_chat:
|
||||
|
|
@ -176,6 +175,8 @@ If you understand, please reply to the following:<|end_of_turn|>
|
|||
return False
|
||||
|
||||
def get_log(self):
|
||||
if self.memory is None:
|
||||
return '[]'
|
||||
ret_messages = []
|
||||
for mes in self.memory.chat_memory:
|
||||
if type(mes) is HumanMessage:
|
||||
|
|
|
|||
Loading…
Reference in New Issue