blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add incremental messages token count (#3890)

Browse Source
This commit is contained in:
Zander Chase 2023-05-01 10:36:54 -07:00 committed by GitHub
parent 52e4fba897
commit 484707ad29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
langchain/experimental/autonomous_agents/autogpt/prompt.py
View File

@ -70,6 +70,7 @@ class AutoGPTPrompt(BaseChatPromptTemplate, BaseModel):
if used_tokens + message_tokens > self.send_token_limit - 1000: if used_tokens + message_tokens > self.send_token_limit - 1000:
break break
historical_messages = [message] + historical_messages historical_messages = [message] + historical_messages
used_tokens += message_tokens
input_message = HumanMessage(content=kwargs["user_input"]) input_message = HumanMessage(content=kwargs["user_input"])
messages: List[BaseMessage] = [base_prompt, time_prompt, memory_message] messages: List[BaseMessage] = [base_prompt, time_prompt, memory_message]
messages += historical_messages messages += historical_messages