merge: dev into branch

Author: Spatika-Ganesh

README.md

@@ -20,7 +20,7 @@ Usage examples are available in the [examples](./examples/) directory.
 import workflowai
 
 wai = workflowai.start(
-    url=..., # defaults to WORKFLOWAI_API_URL env var or https://api.workflowai.ai
+    url=..., # defaults to WORKFLOWAI_API_URL env var or https://api.workflowai.com
     api_key=..., # defaults to WORKFLOWAI_API_KEY env var
 )
 ```

workflowai/__init__.py

@@ -19,7 +19,7 @@ def start(url: Optional[str] = None, api_key: Optional[str] = None) -> Client:
 
     Args:
         url (Optional[str], optional): The API endpoint to use.
-            If not provided, the env variable WORKFLOWAI_API_URL is used. Otherwise defaults to https://api.workflowai.ai
+            If not provided, the env variable WORKFLOWAI_API_URL is used. Otherwise defaults to https://api.workflowai.com
         api_key (Optional[str], optional): _description_. If not provided, the env variable WORKFLOWAI_API_KEY is used.
 
     Returns:

workflowai/core/client/client.py

@@ -1,4 +1,5 @@
 import asyncio
+import importlib.metadata
 import os
 from typing import (
     Any,
@@ -37,10 +38,10 @@ def __init__(self, endpoint: Optional[str] = None, api_key: Optional[str] = None
         self.additional_headers = {
             "x-workflowai-source": "sdk",
             "x-workflowai-language": "python",
-            "x-workflowai-version": "0.1.3" #__version__,
+            "x-workflowai-version": importlib.metadata.version("workflowai"),
         }
         self.api = APIClient(
-            endpoint or os.getenv("WORKFLOWAI_API_URL", "https://api.workflowai.ai"),
+            endpoint or os.getenv("WORKFLOWAI_API_URL", "https://api.workflowai.com"),
             api_key or os.getenv("WORKFLOWAI_API_KEY", ""),
             self.additional_headers
         )

workflowai/core/client/client_test.py

@@ -1,3 +1,4 @@
+import importlib.metadata
 import json
 
 import pytest
@@ -105,6 +106,34 @@ async def test_run_with_env(self, httpx_mock: HTTPXMock, client: Client):
             "stream": False,
             "use_cache": "when_available",
         }
+        
+    async def test_success_with_headers(self, httpx_mock: HTTPXMock, client: Client):
+        httpx_mock.add_response(json=fixtures_json("task_run.json"))
+        task = HelloTask(id="123", schema_id=1)
+
+        task_run = await client.run(task, task_input=HelloTaskInput(name="Alice"))
+
+        assert task_run.id == "8f635b73-f403-47ee-bff9-18320616c6cc"
+
+        reqs = httpx_mock.get_requests()
+        assert len(reqs) == 1
+        assert reqs[0].url == "http://localhost:8000/tasks/123/schemas/1/run"
+        headers = {
+            "x-workflowai-source": "sdk",
+            "x-workflowai-language": "python",
+            "x-workflowai-version": importlib.metadata.version("workflowai"),
+        }
+        
+        body = json.loads(reqs[0].content)
+        assert body == {
+            "task_input": {"name": "Alice"},
+            "group": {"properties": {}},
+            "stream": False,
+            "use_cache": "when_available",
+        }
+        # Check for additional headers
+        for key, value in headers.items():
+            assert reqs[0].headers[key] == value
 
     async def test_run_retries_on_too_many_requests(self, httpx_mock: HTTPXMock, client: Client):
         task = HelloTask(id="123", schema_id=1)