Merge pull request #39 from PromptSail/dev

Ollama handling!
PromptSail · Jul 5, 2024 · 9cc683f · 9cc683f
2 parents 970004b + 4e4eb58
commit 9cc683f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/provider_price_list.json b/backend/provider_price_list.json
@@ -256,7 +256,7 @@
         "model_name": "claude-3.5-sonnet-20240620",
         "provider": "Anthropic",
         "start_date": "",
-        "match_pattern": "(?i)^(claude-3-5-sonnet-20240620)$",
+        "match_pattern": "(?i)^(claude-)(3-5|35|3.5)(-sonnet)(-20240620)$",
         "input_price": 0.003,
         "output_price": 0.015,
         "total_price": 0,

diff --git a/backend/src/transactions/use_cases.py b/backend/src/transactions/use_cases.py
@@ -206,7 +206,7 @@ def store_transaction(
     decoder = response._get_content_decoder()
     buf = b"".join(buffer)
 
-    if "localhost" in str(request.__dict__["url"]):
+    if "localhost" in str(request.__dict__["url"]) or "host.docker.internal" in str(request.__dict__["url"]):
         content = buf.decode("utf-8").split("\n")
         rest, content = content[-2], content[:-2]
         response_content = {