Merge pull request #58 from ollama/mxyng/python-user-agent

python user agent

examples/create/main.py

@@ -0,0 +1,20 @@
+import sys
+
+from ollama import create
+
+
+args = sys.argv[1:]
+if len(args) == 2:
+  # create from local file
+  path = args[1]
+else:
+  print('usage: python main.py <name> <filepath>')
+  sys.exit(1)
+
+# TODO: update to real Modelfile values
+modelfile = f"""
+FROM {path}
+"""
+
+for response in create(model=args[0], modelfile=modelfile, stream=True):
+  print(response['status'])

ollama/_client.py

@@ -3,6 +3,7 @@ import io
 import json
 import httpx
 import binascii
+import platform
 import urllib.parse
 from os import PathLike
 from pathlib import Path
@@ -18,6 +19,13 @@ if sys.version_info < (3, 9):
 else:
   from collections.abc import Iterator, AsyncIterator
 
+from importlib import metadata
+
+try:
+  __version__ = metadata.version('ollama')
+except metadata.PackageNotFoundError:
+  __version__ = '0.0.0'
+
 from ollama._types import Message, Options, RequestError, ResponseError
 
 
@@ -37,10 +45,17 @@ class BaseClient:
     - `timeout`: None
     `kwargs` are passed to the httpx client.
     """
+
+    headers = kwargs.pop('headers', {})
+    headers['Content-Type'] = 'application/json'
+    headers['Accept'] = 'application/json'
+    headers['User-Agent'] = f'ollama-python/{__version__} ({platform.machine()} {platform.system().lower()}) Python/{platform.python_version()}'
+
     self._client = client(
       base_url=_parse_host(host or os.getenv('OLLAMA_HOST')),
       follow_redirects=follow_redirects,
       timeout=timeout,
+      headers=headers,
       **kwargs,
     )