GPTRouter 0.4.5__tar.gz → 0.4.7__tar.gz

{GPTRouter-0.4.5 → GPTRouter-0.4.7}/GPTRouter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.4.5
+Version: 0.4.7
 Summary: A Python package for working with GPTRouter APIs
 Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha

{GPTRouter-0.4.5 → GPTRouter-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.4.5
+Version: 0.4.7
 Summary: A Python package for working with GPTRouter APIs
 Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha

{GPTRouter-0.4.5 → GPTRouter-0.4.7}/gpt_router/models.py

@@ -1,5 +1,6 @@
 from uuid import UUID
 from pydantic import BaseModel, Field
+from enum import Enum
 
 from typing import List, Optional, Dict, Any, Union
 from humps import camelize
@@ -15,6 +16,14 @@ class GPTRouterMetadata(BaseModel):
     created_by_user_id: Optional[str] = None
     appId: Optional[str] = None
 
+class ThinkingType(str, Enum):
+    ENABLED = "enabled"
+    DISABLED = "disabled"
+
+
+class ExtendedThinking(BaseModel):
+    type: ThinkingType
+    budget_tokens: Optional[int] = 1024
 
 class GenerationParams(BaseModel):
     messages: Optional[List[Any]] = None
@@ -38,6 +47,8 @@ class GenerationParams(BaseModel):
     appId: Optional[str] = None
     metadata: Optional[dict] = None
     raw_event: Optional[bool] = None
+    thinking: Optional[ExtendedThinking] = None
+    colasce_into_thinking_tag: Optional[bool] = False
 
 
 class ModelGenerationRequest(BaseModel):

{GPTRouter-0.4.5 → GPTRouter-0.4.7}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name="GPTRouter",
-    version="0.4.5",
+    version="0.4.7",
     description="A Python package for working with GPTRouter APIs",
     long_description=long_description,
     long_description_content_type="text/markdown",