Format Python code with psf/black push

Kav-K · Oct 29, 2023 · 9e5aa46 · 9e5aa46
1 parent 69247ab
commit 9e5aa46
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 6 deletions.
diff --git a/cogs/code_interpreter_service_cog.py b/cogs/code_interpreter_service_cog.py
@@ -486,7 +486,12 @@ async def code_interpreter_chat_command(
 
         llm = ChatOpenAI(model=model, temperature=0, openai_api_key=OPENAI_API_KEY)
 
-        memory = ConversationSummaryBufferMemory(memory_key="memory", return_messages=True, llm=llm, max_token_limit=29000 if "gpt-4" in model else 7500 )
+        memory = ConversationSummaryBufferMemory(
+            memory_key="memory",
+            return_messages=True,
+            llm=llm,
+            max_token_limit=29000 if "gpt-4" in model else 7500,
+        )
 
         agent_kwargs = {
             "extra_prompt_messages": [MessagesPlaceholder(variable_name="memory")],

diff --git a/cogs/search_service_cog.py b/cogs/search_service_cog.py
@@ -29,7 +29,11 @@
     AgentExecutor,
 )
 from langchain.chat_models import ChatOpenAI
-from langchain.memory import ConversationBufferMemory, CombinedMemory, ConversationSummaryBufferMemory
+from langchain.memory import (
+    ConversationBufferMemory,
+    CombinedMemory,
+    ConversationSummaryBufferMemory,
+)
 from langchain.prompts import (
     ChatPromptTemplate,
     SystemMessagePromptTemplate,
@@ -501,8 +505,12 @@ async def search_chat_command(
 
         llm = ChatOpenAI(model=model, temperature=0, openai_api_key=OPENAI_API_KEY)
 
-        memory = ConversationSummaryBufferMemory(memory_key="memory", return_messages=True, llm=llm,
-                                                 max_token_limit=29000 if "gpt-4" in model else 7500)
+        memory = ConversationSummaryBufferMemory(
+            memory_key="memory",
+            return_messages=True,
+            llm=llm,
+            max_token_limit=29000 if "gpt-4" in model else 7500,
+        )
 
         agent_kwargs = {
             "extra_prompt_messages": [MessagesPlaceholder(variable_name="memory")],

diff --git a/models/index_model.py b/models/index_model.py
@@ -333,8 +333,12 @@ async def start_index_chat(self, ctx, search, user, model):
         )
         llm = ChatOpenAI(model=model, temperature=0)
 
-        memory = ConversationSummaryBufferMemory(memory_key="memory", return_messages=True, llm=llm,
-                                                 max_token_limit=29000 if "gpt-4" in model else 7500)
+        memory = ConversationSummaryBufferMemory(
+            memory_key="memory",
+            return_messages=True,
+            llm=llm,
+            max_token_limit=29000 if "gpt-4" in model else 7500,
+        )
 
         agent_chain = create_llama_chat_agent(toolkit, llm, memory=memory, verbose=True)