black formatting

This commit is contained in:
Steve Androulakis
2024-12-31 12:28:02 -08:00
parent c6ea92bfe5
commit 396b748b7d
3 changed files with 20 additions and 20 deletions

View File

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