GPTRouter 0.4.4__tar.gz → 0.4.6__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: GPTRouter
3
- Version: 0.4.4
3
+ Version: 0.4.6
4
4
  Summary: A Python package for working with GPTRouter APIs
5
5
  Author-email: sirjanpreet@writesonic.com
6
6
  Classifier: Development Status :: 3 - Alpha
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: GPTRouter
3
- Version: 0.4.4
3
+ Version: 0.4.6
4
4
  Summary: A Python package for working with GPTRouter APIs
5
5
  Author-email: sirjanpreet@writesonic.com
6
6
  Classifier: Development Status :: 3 - Alpha
@@ -1,7 +1,8 @@
1
1
  from uuid import UUID
2
2
  from pydantic import BaseModel, Field
3
+ from enum import Enum
3
4
 
4
- from typing import List, Optional, Dict, Any
5
+ from typing import List, Optional, Dict, Any, Union
5
6
  from humps import camelize
6
7
 
7
8
 
@@ -15,6 +16,14 @@ class GPTRouterMetadata(BaseModel):
15
16
  created_by_user_id: Optional[str] = None
16
17
  appId: Optional[str] = None
17
18
 
19
+ class ThinkingType(str, Enum):
20
+ ENABLED = "enabled"
21
+ DISABLED = "disabled"
22
+
23
+
24
+ class ExtendedThinking(BaseModel):
25
+ type: ThinkingType
26
+ budget_tokens: Optional[int] = 1024
18
27
 
19
28
  class GenerationParams(BaseModel):
20
29
  messages: Optional[List[Any]] = None
@@ -29,7 +38,7 @@ class GenerationParams(BaseModel):
29
38
  functions: Optional[List[Any]] = None
30
39
  function_call: Optional[dict] = None
31
40
  tools: Optional[List[Any]] = None
32
- tool_choice: Optional[str] = None
41
+ tool_choice: Optional[Union[str, dict]] = None
33
42
  response_format: Optional[Dict[str, Any]] = None
34
43
  parallel_tool_calls: Optional[bool] = None
35
44
  tag: Optional[str] = None
@@ -38,6 +47,7 @@ class GenerationParams(BaseModel):
38
47
  appId: Optional[str] = None
39
48
  metadata: Optional[dict] = None
40
49
  raw_event: Optional[bool] = None
50
+ thinking: Optional[ExtendedThinking] = None
41
51
 
42
52
 
43
53
  class ModelGenerationRequest(BaseModel):
@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
5
5
 
6
6
  setup(
7
7
  name="GPTRouter",
8
- version="0.4.4",
8
+ version="0.4.6",
9
9
  description="A Python package for working with GPTRouter APIs",
10
10
  long_description=long_description,
11
11
  long_description_content_type="text/markdown",
File without changes
File without changes
File without changes
File without changes
File without changes