aixtools 0.3.4__py3-none-any.whl → 0.3.6__py3-none-any.whl

aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.4'
-__version_tuple__ = version_tuple = (0, 3, 4)
+__version__ = version = '0.3.6'
+__version_tuple__ = version_tuple = (0, 3, 6)
 
 __commit_id__ = commit_id = None

aixtools/server/path.py

@@ -36,6 +36,7 @@ def get_workspace_path(ctx: Context | tuple | None = None) -> Path:
     Returns: The workspace path as a Path object.
     """
     user_id, session_id = ctx if isinstance(ctx, tuple) else get_session_id_tuple(ctx)
+    user_id = user_id.lower()  # to handle tuple passed directly
     return WORKSPACES_ROOT_DIR / user_id / session_id
 
 

aixtools/testing/aix_test_model.py

@@ -19,6 +19,20 @@ from ..utils.utils import async_iter
 FINAL_RESULT_TOOL_NAME = "final_result"
 
 
+def _convert_to_parts(items: list[str | TextPart | ToolCallPart]) -> list[TextPart | ToolCallPart]:
+    """Convert a list of strings, TextParts, or ToolCallParts to a list of Parts."""
+    parts: list[TextPart | ToolCallPart] = []
+    for item in items:
+        match item:
+            case str():
+                parts.append(TextPart(item))
+            case TextPart() | ToolCallPart():
+                parts.append(item)
+            case _:
+                raise ValueError(f"Invalid item type in list: {type(item)}, item: {item}")
+    return parts
+
+
 def final_result_tool(result: BaseModel | dict) -> ToolCallPart:
     """Create a ToolCallPart for the final result."""
     if isinstance(result, BaseModel):
@@ -33,7 +47,11 @@ class AixTestModel(Model):
     Test model, returns a specified list of answers, including messages or tool calls
     This is used for testing the agent and model interaction with the rest of the system.
 
-    responses: Is a list of strings or ToolCallPart objects that the model will return in order.
+    responses: Is a list of response items that the model will return in order. Each item can be:
+        - str: A simple text response
+        - TextPart: A text part
+        - ToolCallPart: A tool call
+        - list[str | TextPart | ToolCallPart]: Multiple parts in a single response
 
     Note: The agent will continue to invoke 'request()' until it returns a txt (i.e. not a ToolCallPart).
 
@@ -69,11 +87,25 @@ class AixTestModel(Model):
                     ]
                 )
     ```
+
+    Example with multiple parts in a single response:
+    ```
+        model = AixTestModel(
+                    responses=[
+                        [
+                            "First, let me think about this...",
+                            ToolCallPart(tool_name='analyze', args={'data': 'xyz'}),
+                            TextPart("Analysis complete."),
+                        ],
+                        "Final answer after analysis.",
+                    ]
+                )
+    ```
     """
 
     def __init__(  # pylint: disable=super-init-not-called
         self,
-        responses: list[str | TextPart | ToolCallPart] | AsyncGeneratorType,
+        responses: list[str | TextPart | ToolCallPart | list[str | TextPart | ToolCallPart]] | AsyncGeneratorType,
         sleep_time: float | None = None,
     ):
         self.response_iter = responses(self) if callable(responses) else async_iter(responses)
@@ -141,15 +173,14 @@ class AixTestModel(Model):
         if callable(res):
             res = res(self, messages)
         match res:
-            case str():
-                return ModelResponse(parts=[TextPart(res)], model_name=self.model_name)
-            case ToolCallPart():
-                return ModelResponse(parts=[res], model_name=self.model_name)
-            case TextPart():
-                return ModelResponse(parts=[res], model_name=self.model_name)
             case Exception():
                 raise res
+            case list():
+                parts = _convert_to_parts(res)
+            case str() | TextPart() | ToolCallPart():
+                parts = _convert_to_parts([res])
             case _:
                 raise ValueError(f"Invalid response type: {type(res)}, response: {res}")
         if self.sleep_time:
             await asyncio.sleep(self.sleep_time)
+        return ModelResponse(parts=parts, model_name=self.model_name)

{aixtools-0.3.4.dist-info → aixtools-0.3.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.3.4
+Version: 0.3.6
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown

{aixtools-0.3.4.dist-info → aixtools-0.3.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=3nDaC5e0d_scBB1bUEKPlItbvbY0PmXNNyyOTNFNWNI,704
+aixtools/_version.py,sha256=DPG4bRRG7SLMPQyCzFx7koj75iQndn-UI-ENmNbYVCQ,704
 aixtools/app.py,sha256=JzQ0nrv_bjDQokllIlGHOV0HEb-V8N6k_nGQH-TEsVU,5227
 aixtools/chainlit.md,sha256=yC37Ly57vjKyiIvK4oUvf4DYxZCwH7iocTlx7bLeGLU,761
 aixtools/context.py,sha256=I_MD40ZnvRm5WPKAKqBUAdXIf8YaurkYUUHSVVy-QvU,598
@@ -74,11 +74,11 @@ aixtools/mcp/server.py,sha256=wpN7tJmUZqGsb_rOVfXbyTFJ9VPnQl7nG5wa1JAz-No,2566
 aixtools/model_patch/model_patch.py,sha256=JT-oHubIn2LeoSwWbzEQ5vLH7crJmFUecHyQfaAFHa0,1813
 aixtools/server/__init__.py,sha256=rwPx020YpOzCnrxA80Lc4yLLcIp-Mpe9hNqVO9wDPv0,448
 aixtools/server/app_mounter.py,sha256=0tJ0tC140ezAjnYdlhpLJQjY-TO8NVw7D8LseYCCVY8,3336
-aixtools/server/path.py,sha256=nI4yRQcE6gjKx5GG3PmHf7iT1FelT6Q8Xhw4ol9O1e0,5219
+aixtools/server/path.py,sha256=fvJUmggiOQiIewflYMKZwy5nEJLFGeZHrnRY3xPqHQ0,5284
 aixtools/server/utils.py,sha256=jIRgFT_tdgwuyI8vjc9iQJyDvC0CAoi09Gc5of6I1dg,3554
 aixtools/testing/__init__.py,sha256=mlmaAR2gmS4SbsYNCxnIprmFpFp-syjgVUkpUszo3mE,166
 aixtools/testing/agent_mock.py,sha256=D4DrRGaeKuWEhq8hdB5s9PdfLAXHMKiC3JA7M8yN13o,4244
-aixtools/testing/aix_test_model.py,sha256=i0xBdmpKoEfJHle6JDmcoJLUENN8Eqt181_WZ7XtDdU,6240
+aixtools/testing/aix_test_model.py,sha256=ylmZ-fcG8Y_5lQHzI-B4oQg8Tp6r67EfTqSoy7kyQE4,7429
 aixtools/testing/mock_tool.py,sha256=4I0LxxSkLhGIKM2YxCP3cnYI8IYJjdKhfwGZ3dioXsM,2465
 aixtools/testing/model_patch_cache.py,sha256=238gKC_gSpR3BkeejhetObOkpOR1l2Iz3A6B_eUTRNc,10158
 aixtools/tools/doctor/__init__.py,sha256=FPwYzC1eJyw8IH0-BP0wgxSprLy6Y_4yXCek7496f2k,64
@@ -96,8 +96,8 @@ aixtools/utils/chainlit/cl_agent_show.py,sha256=vaRuowp4BRvhxEr5hw0zHEJ7iaSF_5bo
 aixtools/utils/chainlit/cl_utils.py,sha256=fxaxdkcZg6uHdM8uztxdPowg3a2f7VR7B26VPY4t-3c,5738
 aixtools/vault/__init__.py,sha256=fsr_NuX3GZ9WZ7dGfe0gp_5-z3URxAfwVRXw7Xyc0dU,141
 aixtools/vault/vault.py,sha256=9dZLWdZQk9qN_Q9Djkofw9LUKnJqnrX5H0fGusVLBhA,6037
-aixtools-0.3.4.dist-info/METADATA,sha256=tIIys8pgVbNONjTKYW0fVbbEtkhwQP8F18hYuPQff1c,28014
-aixtools-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.3.4.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
-aixtools-0.3.4.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
-aixtools-0.3.4.dist-info/RECORD,,
+aixtools-0.3.6.dist-info/METADATA,sha256=qi2vJXNlGlV7dBWApxOeQcxigy0iYCS8y_7xG_6VzXA,28014
+aixtools-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aixtools-0.3.6.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
+aixtools-0.3.6.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
+aixtools-0.3.6.dist-info/RECORD,,