tokenator 0.1.12__tar.gz → 0.1.13__tar.gz

{tokenator-0.1.12 → tokenator-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: tokenator
-Version: 0.1.12
+Version: 0.1.13
 Summary: Token usage tracking wrapper for LLMs
 License: MIT
 Author: Ujjwal Maheshwari
@@ -186,7 +186,7 @@ xai_client = OpenAI(
         )
 
 # Wrap it with Tokenator
-client = tokenator_openai(client, db_path=temp_db, provider="xai")
+client = tokenator_openai(xai_client, db_path=temp_db, provider="xai")
 
 # Use it exactly like the OpenAI client but with xAI models
 response = client.chat.completions.create(
@@ -210,17 +210,17 @@ For example, let's see how we can track usage of `perplexity` tokens.
 from openai import OpenAI
 from tokenator import tokenator_openai
 
-xai_client = OpenAI(
+perplexity_client = OpenAI(
             api_key=os.getenv("PERPLEXITY_API_KEY"),
             base_url="https://api.perplexity.ai"
         )
 
 # Wrap it with Tokenator
-client = tokenator_openai(client, db_path=temp_db, provider="perplexity")
+client = tokenator_openai(perplexity_client, db_path=temp_db, provider="perplexity")
 
-# Use it exactly like the OpenAI client but with xAI models
+# Use it exactly like the OpenAI client but with perplexity models
 response = client.chat.completions.create(
-    model="grok-2-latest",
+    model="llama-3.1-sonar-small-128k-online",
     messages=[{"role": "user", "content": "Hello!"}]
 )
 

{tokenator-0.1.12 → tokenator-0.1.13}/README.md

@@ -164,7 +164,7 @@ xai_client = OpenAI(
         )
 
 # Wrap it with Tokenator
-client = tokenator_openai(client, db_path=temp_db, provider="xai")
+client = tokenator_openai(xai_client, db_path=temp_db, provider="xai")
 
 # Use it exactly like the OpenAI client but with xAI models
 response = client.chat.completions.create(
@@ -188,17 +188,17 @@ For example, let's see how we can track usage of `perplexity` tokens.
 from openai import OpenAI
 from tokenator import tokenator_openai
 
-xai_client = OpenAI(
+perplexity_client = OpenAI(
             api_key=os.getenv("PERPLEXITY_API_KEY"),
             base_url="https://api.perplexity.ai"
         )
 
 # Wrap it with Tokenator
-client = tokenator_openai(client, db_path=temp_db, provider="perplexity")
+client = tokenator_openai(perplexity_client, db_path=temp_db, provider="perplexity")
 
-# Use it exactly like the OpenAI client but with xAI models
+# Use it exactly like the OpenAI client but with perplexity models
 response = client.chat.completions.create(
-    model="grok-2-latest",
+    model="llama-3.1-sonar-small-128k-online",
     messages=[{"role": "user", "content": "Hello!"}]
 )
 

{tokenator-0.1.12 → tokenator-0.1.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tokenator"
-version = "0.1.12"
+version = "0.1.13"
 description = "Token usage tracking wrapper for LLMs"
 authors = ["Ujjwal Maheshwari <your.email@example.com>"]
 readme = "README.md"