[PROJ] Format code with Black

Author: actions-user

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/client.py

@@ -396,7 +396,7 @@ def _query(
         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 +440,13 @@ def _query(
 
         # 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 +466,7 @@ async def aquery(
         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 +490,12 @@ async def aquery(
             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 +504,7 @@ def query(
         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 +528,20 @@ def query(
             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 +567,12 @@ def batch_query(
 
         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/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