tokenbee-sdk 0.2.1__tar.gz → 0.2.4__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.4
2
2
  Name: tokenbee-sdk
3
- Version: 0.2.1
3
+ Version: 0.2.4
4
4
  Summary: Official Python SDK for TokenBee LLM inference gateway and observability.
5
5
  Author: TokenBee Inc.
6
6
  Author-email: "TokenBee Inc." <founders@tokenbee.io>
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "tokenbee-sdk"
7
- version = "0.2.1"
7
+ version = "0.2.4"
8
8
  authors = [
9
9
  { name="TokenBee Inc.", email="founders@tokenbee.io" },
10
10
  ]
@@ -6,7 +6,7 @@ long_description = (this_directory / "README.md").read_text()
6
6
 
7
7
  setup(
8
8
  name="tokenbee-sdk",
9
- version="0.2.1",
9
+ version="0.2.4",
10
10
  packages=find_packages(),
11
11
  install_requires=["httpx>=0.23.0"],
12
12
  author="TokenBee Inc.",
@@ -52,7 +52,6 @@ class TokenBee:
52
52
  self,
53
53
  api_key: str,
54
54
  llm_key: str,
55
- base_url: str = "https://api.tokenbee.dev/v1",
56
55
  compression: str = "auto",
57
56
  rate: str = CompressionRate.MEDIUM,
58
57
  model: str = "",
@@ -61,9 +60,11 @@ class TokenBee:
61
60
  ):
62
61
  self.api_key = api_key
63
62
  self.llm_key = llm_key
64
- self.base_url = base_url
63
+ self._base_url = "https://tokenbee.io/v1"
65
64
  self.headers = {
66
65
  "Authorization": f"Bearer {api_key}",
66
+ "X-TokenBee-Key": api_key,
67
+ "X-TB-Key": api_key,
67
68
  "X-LLM-Key": llm_key,
68
69
  "X-TokenBee-Compression": compression,
69
70
  "X-TokenBee-Rate": rate,
@@ -74,7 +75,7 @@ class TokenBee:
74
75
  if provider:
75
76
  self.headers["X-TokenBee-Provider"] = provider
76
77
 
77
- self.client = httpx.Client(headers=self.headers, base_url=self.base_url)
78
+ self.client = httpx.Client(headers=self.headers, base_url=self._base_url)
78
79
 
79
80
  def send(self, model: str, input: dict):
80
81
  parts = model.split("/")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: tokenbee-sdk
3
- Version: 0.2.1
3
+ Version: 0.2.4
4
4
  Summary: Official Python SDK for TokenBee LLM inference gateway and observability.
5
5
  Author: TokenBee Inc.
6
6
  Author-email: "TokenBee Inc." <founders@tokenbee.io>
File without changes
File without changes