pydantic types

ollama · Sep 18, 2024 · 5f4496e · 5f4496e
1 parent 61c8d0d
commit 5f4496e
Show file tree

Hide file tree

Showing 7 changed files with 1,209 additions and 638 deletions.
diff --git a/ollama/__init__.py b/ollama/__init__.py
@@ -1,36 +1,38 @@
 from ollama._client import Client, AsyncClient
 from ollama._types import (
+  Options,
+  Message,
+  Tool,
   GenerateResponse,
   ChatResponse,
+  EmbedResponse,
+  EmbeddingsResponse,
+  StatusResponse,
   ProgressResponse,
-  Message,
-  Options,
+  ListResponse,
+  ShowResponse,
+  ProcessResponse,
   RequestError,
   ResponseError,
 )
 
 __all__ = [
   'Client',
   'AsyncClient',
+  'Options',
+  'Message',
+  'Tool',
   'GenerateResponse',
   'ChatResponse',
+  'EmbedResponse',
+  'EmbeddingsResponse',
+  'StatusResponse',
   'ProgressResponse',
-  'Message',
-  'Options',
+  'ListResponse',
+  'ShowResponse',
+  'ProcessResponse',
   'RequestError',
   'ResponseError',
-  'generate',
-  'chat',
-  'embed',
-  'embeddings',
-  'pull',
-  'push',
-  'create',
-  'delete',
-  'list',
-  'copy',
-  'show',
-  'ps',
 ]
 
 _client = Client()