Fix ollama chat (#10230)

5fc696a2 · Logan · GitHub · f1dc03c0 · 5fc696a2
Unverified Commit 5fc696a2 authored 1 year ago by Logan Committed by GitHub 1 year ago
--- a/llama_index/llms/ollama.py
+++ b/llama_index/llms/ollama.py
@@ -87,7 +87,7 @@ class Ollama(CustomLLM):
            "model": self.model,
            "messages": [
                {
-                    "role": message.role,
+                    "role": message.role.value,
                    "content": message.content,
                    **message.additional_kwargs,
                }
@@ -126,7 +126,7 @@ class Ollama(CustomLLM):
            "model": self.model,
            "messages": [
                {
-                    "role": message.role,
+                    "role": message.role.value,
                    "content": message.content,
                    **message.additional_kwargs,
                }