agentx-python 0.4.12__py3-none-any.whl → 0.4.13__py3-none-any.whl

agentx/agentx.py

@@ -10,7 +10,7 @@ from agentx.resources.workforce import Workforce
 
 class AgentX:
 
-    def __init__(self, api_key: str = None, base_url: str = None):
+    def __init__(self, api_key: str = None, base_url: str = None, workspace_id: str = None):
         self.api_key = api_key or os.getenv("AGENTX_API_KEY")
         if self.api_key and not os.getenv("AGENTX_API_KEY"):
             os.environ["AGENTX_API_KEY"] = self.api_key
@@ -20,6 +20,8 @@ class AgentX:
         if self.base_url:
             os.environ["AGENTX_API_BASE_URL"] = self.base_url
 
+        self.workspace_id = workspace_id or os.getenv("AGENTX_WORKSPACE_ID")
+
         from agentx.evaluations.client import EvaluationsClient
         from agentx.evaluations.runner import EvaluationsRunner
         from agentx.tracing.ingest_client import IngestClient
@@ -37,6 +39,7 @@ class AgentX:
             api_key=self.api_key,
             sdk_version=VERSION,
             base_url=self.base_url,
+            workspace_id=self.workspace_id,
         )
         self.tracer = Tracer(_ingest_client)
 

agentx/tracing/ingest_client.py

@@ -41,10 +41,13 @@ class IngestClient:
         api_key: str,
         sdk_version: str = "unknown",
         base_url: Optional[str] = None,
+        workspace_id: Optional[str] = None,
     ) -> None:
         if not api_key:
             raise ValueError("AGENTX_API_KEY is required")
 
+        self._workspace_id = workspace_id or os.getenv("AGENTX_WORKSPACE_ID")
+
         self._api_key = api_key
         self._sdk_version = sdk_version
 
@@ -76,6 +79,8 @@ class IngestClient:
 
     def enqueue(self, payload: Dict[str, Any]) -> None:
         """Add a trace payload to the send queue. Never blocks; drops silently on overflow."""
+        if self._workspace_id:
+            payload = {**payload, "workspaceId": self._workspace_id}
         try:
             self._queue.put_nowait(payload)
         except queue.Full:
@@ -105,6 +110,8 @@ class IngestClient:
         payload: Dict[str, Any] = {"datasetId": dataset_id}
         if question_index is not None:
             payload["question_index"] = question_index
+        if self._workspace_id:
+            payload["workspaceId"] = self._workspace_id
 
         resp = self._session.post(url, json=payload, timeout=60)
         resp.raise_for_status()
@@ -131,8 +138,9 @@ class IngestClient:
             payload["pass_rate_threshold"] = pass_rate_threshold
         if git_context:
             payload["git_context"] = git_context
-        if workspace_id:
-            payload["workspaceId"] = workspace_id
+        resolved_workspace = workspace_id or self._workspace_id
+        if resolved_workspace:
+            payload["workspaceId"] = resolved_workspace
 
         url = f"{self._base_url}/ingest/ci-runs"
         resp = self._session.post(url, json=payload, timeout=30)
@@ -212,8 +220,9 @@ class IngestClient:
     ) -> Dict[str, Any]:
         """Fetch test case queries for a dataset without creating a CI run."""
         params: Dict[str, str] = {}
-        if workspace_id:
-            params["workspaceId"] = workspace_id
+        resolved_workspace = workspace_id or self._workspace_id
+        if resolved_workspace:
+            params["workspaceId"] = resolved_workspace
         url = f"{self._base_url}/ingest/datasets/{dataset_id}/test-cases"
         resp = self._session.get(url, params=params, timeout=15)
         self._raise_for_ci_status(resp)

agentx/version.py

@@ -1 +1 @@
-VERSION = "0.4.12"
+VERSION = "0.4.13"

{agentx_python-0.4.12.dist-info → agentx_python-0.4.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.4.12
+Version: 0.4.13
 Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python
 Author: Robin Wang and AgentX Team
@@ -14,8 +14,7 @@ License-File: LICENSE
 Requires-Dist: urllib3>=1.26.11
 Requires-Dist: certifi
 Requires-Dist: requests
-Requires-Dist: pydantic
-Requires-Dist: pydantic_core
+Requires-Dist: pydantic>=2.0.0
 Provides-Extra: langchain
 Requires-Dist: langchain-core>=0.1.0; extra == "langchain"
 Provides-Extra: crewai

{agentx_python-0.4.12.dist-info → agentx_python-0.4.13.dist-info}/RECORD

@@ -1,8 +1,8 @@
 agentx/__init__.py,sha256=4iMiGLNU4S-I-WyX0Z7l0BFUY_krVJ3seMQPqyO-ReA,602
-agentx/agentx.py,sha256=rWfeFBS4wV7b9HzJlkO651pgPWo58tZFmPDgPfq6eQo,3321
+agentx/agentx.py,sha256=tw3CUnkvTM1eopoO7u8TgKnAz9t0HNSxrVcjIJsytAg,3469
 agentx/exceptions.py,sha256=TKooc1_pQ2P_4DqpLfZx_ap-6cfrjFLW8JS2qaGx_1M,1237
 agentx/util.py,sha256=yOV3VVdc0KUbw5pEkSRiO2lZzEtL1vOsnB_f5tFk948,665
-agentx/version.py,sha256=M4jHf2piUpaEzqXOe34S2F5941xUUdbYw1L0mpXlyj8,19
+agentx/version.py,sha256=dO83YWKEtJW3A7sv1BcvYqTJgQZCIf7sMXiONZFOY_Q,19
 agentx/evaluations/__init__.py,sha256=OmdCHiBO3Qcsck0UkOyVqSvfRs6nXbSi-2FXzyRfsog,89
 agentx/evaluations/_term.py,sha256=S4blZgH66wHvkt2hR29h8JE5xnakBeD7dlZdUW9Wfoo,2065
 agentx/evaluations/client.py,sha256=3mC0uSBRk9uO4F3a7kIMM_UUsyypF99_97E5MSpYFzM,7736
@@ -28,10 +28,10 @@ agentx/resources/conversation.py,sha256=HK7gEGK6qIUz0KvG-ItUip08otWZdkP5S6qHiznv
 agentx/resources/workforce.py,sha256=lfGVkoV9lcOp-lZScjTvwRarJMkhzfMfLm4syJDujGc,4168
 agentx/tracing/__init__.py,sha256=l2mIRILE-wWrCD0fekgrIOQUEvtDObzdrkgpQUtXroU,408
 agentx/tracing/ci_types.py,sha256=zFVcZGvc1qbVDdOlEPQ1i5R6terqJHxCzFF4ZjouxxI,1423
-agentx/tracing/ingest_client.py,sha256=p-rqzYEvFgUwRIcY54omXz4tPvw-hHzysVCCA2PSpuo,11025
+agentx/tracing/ingest_client.py,sha256=8aLabvWecgZsn0pkajZkI1it2TOk5boP0kC3o6TO52o,11487
 agentx/tracing/tracer.py,sha256=nJILoeQg6HV4SR05xJPy6EiSiZCpkp4yFBTrN1BO_1w,16951
-agentx_python-0.4.12.dist-info/licenses/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
-agentx_python-0.4.12.dist-info/METADATA,sha256=l_nlLJPCx7NjWvDOlJXEmDlG8m0U9QxJ1eJBOkPDg1U,6652
-agentx_python-0.4.12.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-agentx_python-0.4.12.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
-agentx_python-0.4.12.dist-info/RECORD,,
+agentx_python-0.4.13.dist-info/licenses/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
+agentx_python-0.4.13.dist-info/METADATA,sha256=nDiVYt09aeyL89w3-ro9WfjsD7abgG0DdyujwsOdC5Q,6630
+agentx_python-0.4.13.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agentx_python-0.4.13.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.4.13.dist-info/RECORD,,