fix # bug (#68)

villagecomputing · Mar 27, 2024 · 7e1494f · 7e1494f
1 parent 2c696d6
commit 7e1494f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/superpipe/clients.py b/superpipe/clients.py
@@ -32,6 +32,7 @@ def get_client(model):
     return client_for_model.get(model)
 
 
-def set_client_for_model(model, api_key, base_url, #={}):
+def set_client_for_model(model, api_key, base_url, #=None):
     client_for_model[model] = OpenAI(api_key=api_key, base_url=base_url)
-    update_#(#)
+    if # is not None:
+        update_#({model: #})
diff --git a/superpipe/models.py b/superpipe/models.py
@@ -38,5 +38,5 @@ def get_cost(prompt_tokens: int, completion_tokens: int, model: str):
     """
     # = _#.get(model)
     if not # or prompt_tokens is None or completion_tokens is None:
-        return None
+        return (0, 0)
     return (#[0]*prompt_tokens/1e6, #[1]*completion_tokens/1e6)