Merge pull request #465 from macrocosm-os/staging

Update SN19 JSON (#464)
macrocosm-os · Nov 26, 2024 · d497ee6 · d497ee6
2 parents 7497354 + ab77640
commit d497ee6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/prompting/llms/apis/sn19_wrapper.py b/prompting/llms/apis/sn19_wrapper.py
@@ -38,6 +38,7 @@ def chat_complete(
     }
     response = requests.post(url, headers=headers, data=json.dumps(data))
     try:
-        return response.json()["choices"][0]["delta"]["content"]
+        response_json = response.json()
+        return response_json["choices"][0]["message"].get("content")
     except Exception as e:
         logger.exception(f"Error in chat_complete: {e}")
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prompting"
-version = "2.13.0"
+version = "2.13.1"
 description = "Subnetwork 1 runs on Bittensor and is maintained by Macrocosmos. It's an effort to create decentralised AI"
 authors = ["Kalei Brady, Dmytro Bobrenko, Felix Quinque, Steffen Cruz"]
 readme = "README.md"