weco 0.1.8__tar.gz → 0.1.9__tar.gz

{weco-0.1.8 → weco-0.1.9}/.github/workflows/release.yml

@@ -79,7 +79,7 @@ jobs:
         GITHUB_TOKEN: ${{ github.token }}
       run: >-
         gh release create
-        'v0.1.8'
+        'v0.1.9'
         --repo '${{ github.repository }}'
         --notes ""
     - name: Upload artifact signatures to GitHub Release
@@ -90,5 +90,5 @@ jobs:
       # sigstore-produced signatures and certificates.
       run: >-
         gh release upload
-        'v0.1.8' dist/**
+        'v0.1.9' dist/**
         --repo '${{ github.repository }}'

{weco-0.1.8 → weco-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: weco
-Version: 0.1.8
+Version: 0.1.9
 Summary: A client facing API for interacting with the WeCo AI function builder service.
 Author-email: WeCo AI Team <dhruv@weco.ai>
 License: MIT

{weco-0.1.8 → weco-0.1.9}/pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 ]
 description = "A client facing API for interacting with the WeCo AI function builder service."
 readme = "README.md"
-version = "0.1.8"
+version = "0.1.9"
 license = {text = "MIT"}
 requires-python = ">=3.8"
 dependencies = ["asyncio", "httpx[http2]", "pillow"]

{weco-0.1.8 → weco-0.1.9}/tests/test_reasoning.py

@@ -7,7 +7,8 @@ def assert_query_response(query_response):
     assert isinstance(query_response, dict)
     assert isinstance(query_response["output"], dict)
     assert isinstance(query_response["reasoning_steps"], list)
-    for step in query_response["reasoning_steps"]: assert isinstance(step, str)
+    for step in query_response["reasoning_steps"]:
+        assert isinstance(step, str)
     assert isinstance(query_response["in_tokens"], int)
     assert isinstance(query_response["out_tokens"], int)
     assert isinstance(query_response["latency_ms"], float)
@@ -24,11 +25,14 @@ def text_reasoning_evaluator():
 
 def test_text_reasoning_query(text_reasoning_evaluator):
     fn_name, version_number, _ = text_reasoning_evaluator
-    query_response = query(fn_name=fn_name, version_number=version_number, text_input="I love this product!", return_reasoning=True)
+    query_response = query(
+        fn_name=fn_name, version_number=version_number, text_input="I love this product!", return_reasoning=True
+    )
 
     assert_query_response(query_response)
     assert set(query_response["output"].keys()) == {"sentiment", "explanation"}
 
+
 @pytest.fixture
 def vision_reasoning_evaluator():
     fn_name, version_number, fn_desc = build(

{weco-0.1.8 → weco-0.1.9}/weco/client.py

@@ -68,6 +68,7 @@ class WecoAI:
         self.http2 = http2
         self.timeout = timeout
         self.base_url = "https://function.api.weco.ai"
+
         # Setup clients
         self.client = httpx.Client(http2=http2, timeout=timeout)
         self.async_client = httpx.AsyncClient(http2=http2, timeout=timeout)
@@ -396,7 +397,7 @@ class WecoAI:
         version_number: Optional[int],
         text_input: Optional[str],
         images_input: Optional[List[str]],
-        return_reasoning: Optional[bool]
+        return_reasoning: Optional[bool],
     ) -> Union[Dict[str, Any], Coroutine[Any, Any, Dict[str, Any]]]:
         """Internal method to handle both synchronous and asynchronous query requests.
 
@@ -440,7 +441,13 @@ class WecoAI:
 
         # Make the request
         endpoint = "query"
-        data = {"name": fn_name, "text": text_input, "images": image_urls, "version_number": version_number, "return_reasoning": return_reasoning}
+        data = {
+            "name": fn_name,
+            "text": text_input,
+            "images": image_urls,
+            "version_number": version_number,
+            "return_reasoning": return_reasoning,
+        }
         request = self._make_request(endpoint=endpoint, data=data, is_async=is_async)
 
         if is_async:
@@ -460,7 +467,7 @@ class WecoAI:
         version_number: Optional[int] = -1,
         text_input: Optional[str] = "",
         images_input: Optional[List[str]] = [],
-        return_reasoning: Optional[bool] = False
+        return_reasoning: Optional[bool] = False,
     ) -> Dict[str, Any]:
         """Asynchronously queries a function with the given function ID and input.
 
@@ -484,7 +491,12 @@ class WecoAI:
             and the latency in milliseconds.
         """
         return await self._query(
-            fn_name=fn_name, version_number=version_number, text_input=text_input, images_input=images_input, return_reasoning=return_reasoning, is_async=True
+            fn_name=fn_name,
+            version_number=version_number,
+            text_input=text_input,
+            images_input=images_input,
+            return_reasoning=return_reasoning,
+            is_async=True,
         )
 
     def query(
@@ -493,7 +505,7 @@ class WecoAI:
         version_number: Optional[int] = -1,
         text_input: Optional[str] = "",
         images_input: Optional[List[str]] = [],
-        return_reasoning: Optional[bool] = False
+        return_reasoning: Optional[bool] = False,
     ) -> Dict[str, Any]:
         """Synchronously queries a function with the given function ID and input.
 
@@ -517,11 +529,20 @@ class WecoAI:
             and the latency in milliseconds.
         """
         return self._query(
-            fn_name=fn_name, version_number=version_number, text_input=text_input, images_input=images_input, return_reasoning=return_reasoning, is_async=False
+            fn_name=fn_name,
+            version_number=version_number,
+            text_input=text_input,
+            images_input=images_input,
+            return_reasoning=return_reasoning,
+            is_async=False,
         )
 
     def batch_query(
-        self, fn_name: str, batch_inputs: List[Dict[str, Any]], version_number: Optional[int] = -1, return_reasoning: Optional[bool] = False
+        self,
+        fn_name: str,
+        batch_inputs: List[Dict[str, Any]],
+        version_number: Optional[int] = -1,
+        return_reasoning: Optional[bool] = False,
     ) -> List[Dict[str, Any]]:
         """Batch queries a function version with a list of inputs.
 
@@ -547,7 +568,12 @@ class WecoAI:
 
         async def run_queries():
             tasks = list(
-                map(lambda fn_input: self.aquery(fn_name=fn_name, version_number=version_number, return_reasoning=return_reasoning, **fn_input), batch_inputs)
+                map(
+                    lambda fn_input: self.aquery(
+                        fn_name=fn_name, version_number=version_number, return_reasoning=return_reasoning, **fn_input
+                    ),
+                    batch_inputs,
+                )
             )
             return await asyncio.gather(*tasks)
 

{weco-0.1.8 → weco-0.1.9}/weco/functional.py

@@ -79,7 +79,13 @@ def query(
         and the latency in milliseconds.
     """
     client = WecoAI(api_key=api_key)
-    response = client.query(fn_name=fn_name, version_number=version_number, text_input=text_input, images_input=images_input, return_reasoning=return_reasoning)
+    response = client.query(
+        fn_name=fn_name,
+        version_number=version_number,
+        text_input=text_input,
+        images_input=images_input,
+        return_reasoning=return_reasoning,
+    )
     return response
 
 
@@ -116,13 +122,21 @@ async def aquery(
     """
     client = WecoAI(api_key=api_key)
     response = await client.aquery(
-        fn_name=fn_name, version_number=version_number, text_input=text_input, images_input=images_input, return_reasoning=return_reasoning
+        fn_name=fn_name,
+        version_number=version_number,
+        text_input=text_input,
+        images_input=images_input,
+        return_reasoning=return_reasoning,
     )
     return response
 
 
 def batch_query(
-    fn_name: str, batch_inputs: List[Dict[str, Any]], version_number: Optional[int] = -1, return_reasoning: Optional[bool] = False, api_key: Optional[str] = None
+    fn_name: str,
+    batch_inputs: List[Dict[str, Any]],
+    version_number: Optional[int] = -1,
+    return_reasoning: Optional[bool] = False,
+    api_key: Optional[str] = None,
 ) -> List[Dict[str, Any]]:
     """Synchronously queries multiple functions using asynchronous calls internally.
 
@@ -153,5 +167,7 @@ def batch_query(
         in the same order as the input queries.
     """
     client = WecoAI(api_key=api_key)
-    responses = client.batch_query(fn_name=fn_name, version_number=version_number, batch_inputs=batch_inputs, return_reasoning=return_reasoning)
+    responses = client.batch_query(
+        fn_name=fn_name, version_number=version_number, batch_inputs=batch_inputs, return_reasoning=return_reasoning
+    )
     return responses

{weco-0.1.8 → weco-0.1.9}/weco.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: weco
-Version: 0.1.8
+Version: 0.1.9
 Summary: A client facing API for interacting with the WeCo AI function builder service.
 Author-email: WeCo AI Team <dhruv@weco.ai>
 License: MIT