(self, messages)
| 18 | return response.message.content |
| 19 | |
| 20 | def chat_stream(self, messages): |
| 21 | response = ollama_chat( |
| 22 | model=self.model, |
| 23 | messages=messages, |
| 24 | stream=True, |
| 25 | ) |
| 26 | for chunk in response: |
| 27 | if chunk.message and chunk.message.content: |
| 28 | yield chunk.message.content |
| 29 | |
| 30 | class LLMHelper: |
| 31 | def __init__(self, logger): |