examples

ollama · Dec 22, 2023 · 0fb7c9f · 0fb7c9f
1 parent 916fe11
commit 0fb7c9f
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 6 deletions.
diff --git a/examples/simple-chat-stream/main.py b/examples/simple-chat-stream/main.py
@@ -8,10 +8,8 @@
   },
 ]
 
-for message in chat('mistral', messages=messages, stream=True):
-  if message := message.get('message'):
-    if message.get('role') == 'assistant':
-      print(message.get('content', ''), end='', flush=True)
+for part in chat('mistral', messages=messages, stream=True):
+  print(part['message']['content'], end='', flush=True)
 
 # end with a newline
 print()
diff --git a/examples/simple-chat/main.py b/examples/simple-chat/main.py
@@ -9,4 +9,4 @@
 ]
 
 response = chat('mistral', messages=messages)
-print(response['message'])
+print(response['message']['content'])
diff --git a/examples/simple-generate-stream/main.py b/examples/simple-generate-stream/main.py
@@ -0,0 +1,5 @@
+from ollama import generate
+
+
+for part in generate('mistral', 'Why is the sky blue?', stream=True):
+  print(part['response'], end='', flush=True)
diff --git a/examples/simple-generate/main.py b/examples/simple-generate/main.py
@@ -0,0 +1,5 @@
+from ollama import generate
+
+
+response = generate('mistral', 'Why is the sky blue?')
+print(response['response'])
diff --git a/examples/simple-multimodal/main.py b/examples/simple-multimodal/main.py
@@ -24,6 +24,6 @@
 raw.raise_for_status()
 
 for response in generate('llava', 'explain this comic:', images=[raw.content], stream=True):
-  print(response.get('response'), end='', flush=True)
+  print(response['response'], end='', flush=True)
 
 print()