machine-thinking 0.0.4__tar.gz → 0.0.6__tar.gz

{machine_thinking-0.0.4 → machine_thinking-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: machine-thinking
-Version: 0.0.4
+Version: 0.0.6
 Summary: Calling the API of 'Thinking-Machines Lab' models without dependencies.
 Author-email: Machina Ratiocinatrix <machina.ratio@gmail.com>, Alexander Fedotov <alex.fedotov@aol.com>
 Project-URL: Homepage, https://github.com/machina-ratiocinatrix/machine-thinking

{machine_thinking-0.0.4 → machine_thinking-0.0.6}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools>=67.0"]
 build-backend = "setuptools.build_meta"
 [project]
 name = "machine-thinking"
-version = "0.0.4"
+version = "0.0.6"
 authors = [
     {name="Machina Ratiocinatrix", email="machina.ratio@gmail.com"},
     {name="Alexander Fedotov", email="alex.fedotov@aol.com"}

{machine_thinking-0.0.4 → machine_thinking-0.0.6}/src/machine_thinking/messages.py

@@ -9,7 +9,8 @@ from .utils import (query,
                     default_model,
                     get_function,
                     get_func_args,
-                    call_function)
+                    call_function,
+                    api_base_ant)
 
 
 def get_weather(location):
@@ -45,7 +46,7 @@ def message(messages=None, instructions=None, tools=None, **kwargs):
         payload['tool_choice'] = 'auto'
 
     while True:
-        result = query(payload, '/messages')
+        result = query(payload, '/messages', api_base_ant)
         completion_message = result['choices'][0]['message']
         messages.append(completion_message)
         thoughts = completion_message.get('reasoning_content', '')

{machine_thinking-0.0.4 → machine_thinking-0.0.6}/src/machine_thinking/utils.py

@@ -13,7 +13,8 @@ from os import environ
 
 api_key = environ.get("TINKER_API_KEY", '')
 default_model = environ.get("TINKER_DEFAULT_MODEL", 'thinkingmachines/Inkling')
-api_base = environ.get("TINKER_API_BASE", 'https://tinker.thinkingmachines.dev/services/tinker-prod/oai/api/v1')
+api_base_oai = environ.get("TINKER_OAI_API_BASE", 'https://tinker.thinkingmachines.dev/services/tinker-prod/oai/api/v1')
+api_base_ant = environ.get("TINKER_ANT_API_BASE", 'https://tinker.thinkingmachines.dev/services/tinker-prod/anthropic/api/v1')
 
 
 # Set the mandatory headers
@@ -72,8 +73,12 @@ def call_function(func, func_args):
     return result
 
 
-def query(payload, url_suffix):
+def query(payload, url_suffix, url_prefix=None):
     # Convert data dictionary to JSON and encode it to bytes
+    if url_prefix:
+        api_base = url_prefix
+    else:
+        api_base= api_base_oai
     data_bytes = json.dumps(payload).encode('utf-8')
     # Create the Request object
     req = urllib.request.Request(

{machine_thinking-0.0.4 → machine_thinking-0.0.6}/src/machine_thinking.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: machine-thinking
-Version: 0.0.4
+Version: 0.0.6
 Summary: Calling the API of 'Thinking-Machines Lab' models without dependencies.
 Author-email: Machina Ratiocinatrix <machina.ratio@gmail.com>, Alexander Fedotov <alex.fedotov@aol.com>
 Project-URL: Homepage, https://github.com/machina-ratiocinatrix/machine-thinking