Feat/wrapper

portkey_ai/__init__.py

@@ -10,23 +10,40 @@
     CacheLiteral,
     Message,
     PortkeyResponse,
-    ChatCompletions,
     Completion,
     AsyncCompletion,
     Params,
     Config,
     RetrySettings,
     ChatCompletion,
     AsyncChatCompletion,
-    ChatCompletionChunk,
-    TextCompletion,
-    TextCompletionChunk,
     createHeaders,
     Prompts,
     AsyncPrompts,
     Portkey,
     AsyncPortkey,
+    Images,
+    AsyncImages,
+    Assistants,
+    AsyncAssistants,
+    Threads,
+    AsyncThreads,
+    Messages,
+    AsyncMessages,
+    MainFiles,
+    AsyncMainFiles,
+    Models,
+    AsyncModels,
+    ThreadFiles,
+    AsyncThreadFiles,
+    AssistantFiles,
+    AsyncAssistantFiles,
+    Runs,
+    AsyncRuns,
+    Steps,
+    AsyncSteps,
 )
+
 from portkey_ai.version import VERSION
 from portkey_ai.api_resources.global_constants import (
     PORTKEY_BASE_URL,
@@ -51,16 +68,12 @@
     "CacheType",
     "CacheLiteral",
     "Message",
-    "ChatCompletions",
     "Completion",
     "AsyncCompletion",
     "Params",
     "RetrySettings",
     "ChatCompletion",
     "AsyncChatCompletion",
-    "ChatCompletionChunk",
-    "TextCompletion",
-    "TextCompletionChunk",
     "Config",
     "api_key",
     "base_url",
@@ -70,4 +83,24 @@
     "AsyncPrompts",
     "Portkey",
     "AsyncPortkey",
+    "Images",
+    "AsyncImages",
+    "Assistants",
+    "AsyncAssistants",
+    "Threads",
+    "AsyncThreads",
+    "Messages",
+    "AsyncMessages",
+    "MainFiles",
+    "AsyncMainFiles",
+    "Models",
+    "AsyncModels",
+    "ThreadFiles",
+    "AsyncThreadFiles",
+    "AssistantFiles",
+    "AsyncAssistantFiles",
+    "Runs",
+    "AsyncRuns",
+    "Steps",
+    "AsyncSteps",
 ]

portkey_ai/api_resources/__init__.py

@@ -11,6 +11,26 @@
     Feedback,
     AsyncFeedback,
     createHeaders,
+    Images,
+    AsyncImages,
+    Assistants,
+    AsyncAssistants,
+    Threads,
+    AsyncThreads,
+    Messages,
+    AsyncMessages,
+    MainFiles,
+    AsyncMainFiles,
+    Models,
+    AsyncModels,
+    ThreadFiles,
+    AsyncThreadFiles,
+    AssistantFiles,
+    AsyncAssistantFiles,
+    Runs,
+    AsyncRuns,
+    Steps,
+    AsyncSteps,
 )
 from .utils import (
     Modes,
@@ -25,10 +45,6 @@
     Params,
     Config,
     RetrySettings,
-    ChatCompletions,
-    ChatCompletionChunk,
-    TextCompletion,
-    TextCompletionChunk,
 )
 from .client import Portkey, AsyncPortkey
 
@@ -45,17 +61,13 @@
     "CacheType",
     "CacheLiteral",
     "Message",
-    "ChatCompletions",
     "Completion",
     "AsyncCompletion",
     "Params",
     "Config",
     "RetrySettings",
     "ChatCompletion",
     "AsyncChatCompletion",
-    "ChatCompletionChunk",
-    "TextCompletion",
-    "TextCompletionChunk",
     "Generations",
     "AsyncGenerations",
     "Prompts",
@@ -65,4 +77,24 @@
     "createHeaders",
     "Portkey",
     "AsyncPortkey",
+    "Images",
+    "AsyncImages",
+    "Assistants",
+    "AsyncAssistants",
+    "Threads",
+    "AsyncThreads",
+    "Messages",
+    "AsyncMessages",
+    "MainFiles",
+    "AsyncMainFiles",
+    "Models",
+    "AsyncModels",
+    "ThreadFiles",
+    "AsyncThreadFiles",
+    "AssistantFiles",
+    "AsyncAssistantFiles",
+    "Runs",
+    "AsyncRuns",
+    "Steps",
+    "AsyncSteps",
 ]

portkey_ai/api_resources/apis/__init__.py

@@ -5,6 +5,22 @@
 from .create_headers import createHeaders
 from .post import Post, AsyncPost
 from .embeddings import Embeddings, AsyncEmbeddings
+from .images import Images, AsyncImages
+from .assistants import Assistants, AssistantFiles, AsyncAssistants, AsyncAssistantFiles
+from .threads import (
+    Threads,
+    Messages,
+    ThreadFiles,
+    Runs,
+    Steps,
+    AsyncThreads,
+    AsyncMessages,
+    AsyncThreadFiles,
+    AsyncRuns,
+    AsyncSteps,
+)
+from .main_files import MainFiles, AsyncMainFiles
+from .models import Models, AsyncModels
 
 __all__ = [
     "Completion",
@@ -22,4 +38,24 @@
     "AsyncPost",
     "Embeddings",
     "AsyncEmbeddings",
+    "Images",
+    "AsyncImages",
+    "Assistants",
+    "AsyncAssistants",
+    "MainFiles",
+    "AsyncMainFiles",
+    "Models",
+    "AsyncModels",
+    "AssistantFiles",
+    "ThreadFiles",
+    "AsyncAssistantFiles",
+    "AsyncThreadFiles",
+    "Threads",
+    "AsyncThreads",
+    "Messages",
+    "AsyncMessages",
+    "Runs",
+    "AsyncRuns",
+    "Steps",
+    "AsyncSteps",
 ]