From 396b748b7d6223cc8d23b353f3964247fad502eb Mon Sep 17 00:00:00 2001
From: Steve Androulakis <steve.androulakis@gmail.com>
Date: Tue, 31 Dec 2024 12:28:02 -0800
Subject: [PATCH] black formatting

---
 activities.py | 14 ++++++++------
 run_ollama.py | 18 ++++++++++--------
 workflows.py  |  8 ++------
 3 files changed, 20 insertions(+), 20 deletions(-)

diff --git a/activities.py b/activities.py
index 30c3b9a..46a82a5 100644
--- a/activities.py
+++ b/activities.py
@@ -2,24 +2,26 @@ from dataclasses import dataclass
 from temporalio import activity
 from ollama import chat, ChatResponse
 
+
 @dataclass
 class OllamaPromptInput:
     prompt: str
     context_instructions: str
 
+
 class OllamaActivities:
     @activity.defn
     def prompt_ollama(self, input: OllamaPromptInput) -> str:
-        model_name = 'mistral'
+        model_name = "mistral"
         messages = [
             {
-                'role': 'system',
-                'content': input.context_instructions,
+                "role": "system",
+                "content": input.context_instructions,
             },
             {
-                'role': 'user',
-                'content': input.prompt,
-            }
+                "role": "user",
+                "content": input.prompt,
+            },
         ]
 
         response: ChatResponse = chat(model=model_name, messages=messages)
diff --git a/run_ollama.py b/run_ollama.py
index 97a2e6e..310d9e0 100644
--- a/run_ollama.py
+++ b/run_ollama.py
@@ -1,21 +1,23 @@
 from ollama import chat, ChatResponse
 
+
 def main():
-    model_name = 'mistral'
-    
+    model_name = "mistral"
+
     # The messages to pass to the model
     messages = [
         {
-            'role': 'user',
-            'content': 'Why is the sky blue?',
+            "role": "user",
+            "content": "Why is the sky blue?",
         }
     ]
-    
+
     # Call ollama's chat function
     response: ChatResponse = chat(model=model_name, messages=messages)
-    
+
     # Print the full message content
     print(response.message.content)
 
-if __name__ == '__main__':
-    main()
\ No newline at end of file
+
+if __name__ == "__main__":
+    main()
diff --git a/workflows.py b/workflows.py
index a427f65..0dc3427 100644
--- a/workflows.py
+++ b/workflows.py
@@ -158,10 +158,6 @@ class EntityOllamaWorkflow:
     # Return (context_instructions, prompt) for summarizing the conversation
     def prompt_summary_with_history(self) -> tuple[str, str]:
         history_string = self.format_history()
-        context_instructions = (
-            f"Here is the conversation history between a user and a chatbot: {history_string}"
-        )
-        actual_prompt = (
-            "Please produce a two sentence summary of this conversation."
-        )
+        context_instructions = f"Here is the conversation history between a user and a chatbot: {history_string}"
+        actual_prompt = "Please produce a two sentence summary of this conversation."
         return (context_instructions, actual_prompt)