initial commit

examples/simple-chat-stream/main.py

@@ -0,0 +1,17 @@
+from ollama import chat
+
+
+messages = [
+  {
+    'role': 'user',
+    'content': 'Why is the sky blue?',
+  },
+]
+
+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)
+
+# end with a newline
+print()

examples/simple-chat/main.py

@@ -0,0 +1,12 @@
+from ollama import chat
+
+
+messages = [
+  {
+    'role': 'user',
+    'content': 'Why is the sky blue?',
+  },
+]
+
+response = chat('mistral', messages=messages)
+print(response['message'])

examples/simple-multimodal/main.py

@@ -0,0 +1,29 @@
+import sys
+import random
+import httpx
+
+from ollama import generate
+
+
+latest = httpx.get('https://xkcd.com/info.0.json')
+latest.raise_for_status()
+
+if len(sys.argv) > 1:
+  num = int(sys.argv[1])
+else:
+  num = random.randint(1, latest.json().get('num'))
+
+comic = httpx.get(f'https://xkcd.com/{num}/info.0.json')
+comic.raise_for_status()
+
+print(f'xkcd #{comic.json().get("num")}: {comic.json().get("alt")}')
+print(f'link: https://xkcd.com/{num}')
+print('---')
+
+raw = httpx.get(comic.json().get('img'))
+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()