エラー修正

langchain
NON906 2024-01-10 16:24:46 +09:00
parent 9db4b1477b
commit 37927b9fb9
1 changed files with 10 additions and 9 deletions

View File

@ -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: