agenta 0.13.0a1__py3-none-any.whl → 0.13.0a2__py3-none-any.whl

agenta/client/backend/types/create_span.py

@@ -18,6 +18,7 @@ class CreateSpan(pydantic.BaseModel):
     id: str
     app_id: typing.Optional[str]
     variant_id: typing.Optional[str]
+    variant_name: typing.Optional[str]
     inputs: typing.Optional[typing.Dict[str, typing.Any]]
     outputs: typing.Optional[typing.List[str]]
     config: typing.Optional[typing.Dict[str, typing.Any]]

agenta/client/backend/types/get_config_response.py

@@ -12,7 +12,6 @@ except ImportError:
 
 
 class GetConfigResponse(pydantic.BaseModel):
-    config_id: typing.Optional[str]
     config_name: str
     current_version: int
     parameters: typing.Dict[str, typing.Any]

agenta/sdk/agenta_init.py

@@ -90,6 +90,7 @@ class AgentaSingleton:
         self.host = host
         self.app_id = os.environ.get("AGENTA_APP_ID") if app_id is None else app_id
         self.variant_id = os.environ.get("AGENTA_VARIANT_ID")
+        self.variant_name = os.environ.get("AGENTA_VARIANT_NAME")
         self.api_key = api_key
         self.config = Config(base_id=base_id, host=host)
 
@@ -193,7 +194,7 @@ class Config:
                     + str(ex)
                 )
         try:
-            self.set(**config.parameters)
+            self.set(**{"current_version": config.current_version, **config.parameters})
         except Exception as ex:
             logger.warning("Failed to set the configuration with error: " + str(ex))
 
@@ -237,6 +238,7 @@ def llm_tracing(max_workers: Optional[int] = None) -> Tracing:
         base_url=singleton.host,
         app_id=singleton.app_id,  # type: ignore
         variant_id=singleton.variant_id,  # type: ignore
+        variant_name=singleton.variant_name,
         api_key=singleton.api_key,
         max_workers=max_workers,
     )

agenta/sdk/tracing/llm_tracing.py

@@ -36,6 +36,7 @@ class Tracing(object):
         base_url: str,
         app_id: str,
         variant_id: str,
+        variant_name: Optional[str] = None,
         api_key: Optional[str] = None,
         max_workers: Optional[int] = None,
     ):
@@ -44,6 +45,7 @@ class Tracing(object):
         self.llm_logger = llm_logger
         self.app_id = app_id
         self.variant_id = variant_id
+        self.variant_name = variant_name
         self.tasks_manager = TaskQueue(
             max_workers if max_workers else 4, logger=llm_logger
         )
@@ -101,6 +103,7 @@ class Tracing(object):
             id=span_id,
             app_id=self.app_id,
             variant_id=self.variant_id,
+            variant_name=self.variant_name,
             inputs=inputs,
             name=name,
             config=config,
@@ -131,6 +134,7 @@ class Tracing(object):
             name=name,
             app_id=self.app_id,
             variant_id=self.variant_id,
+            variant_name=self.variant_name,
             config=config,
             environment=self.active_trace.environment,
             parent_span_id=self.parent_span_id,

{agenta-0.13.0a1.dist-info → agenta-0.13.0a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.13.0a1
+Version: 0.13.0a2
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering

{agenta-0.13.0a1.dist-info → agenta-0.13.0a2.dist-info}/RECORD

@@ -54,7 +54,7 @@ agenta/client/backend/types/base_output.py,sha256=ynXhDBQKrkR6Lnkx-yv6Q8xW4wXmzX
 agenta/client/backend/types/body_import_testset.py,sha256=7dVF3mv3VO0Co8F0qxLAgu4jabqDPjebK4mYvcd_TuA,1061
 agenta/client/backend/types/config_db.py,sha256=P0cSYvVOn0ZxpYMIdvhWpQVjRuBS5APe6qlc69AXaF4,1028
 agenta/client/backend/types/create_app_output.py,sha256=pgnTnfZx35Q-8wZ1yTZBQ0ydYacGzFC9kyLug_UvymM,986
-agenta/client/backend/types/create_span.py,sha256=q6FDyuS9mMBffcJD7RuYJR9etX9KvaODRW093lQKX1o,1754
+agenta/client/backend/types/create_span.py,sha256=_3G4WUnmgXtTCkIJFpce-9PDh6KDDZ2YkvXVOE8GSx8,1793
 agenta/client/backend/types/create_trace_response.py,sha256=FO-Ii9JEn2AQ1nmZYmjnKRbACsNxRvY_-xn7Ys7Yo8A,1012
 agenta/client/backend/types/docker_env_vars.py,sha256=altCvA1k-zdAkKNYLwaCnmV48HZg9cwe2cHu_BGImac,986
 agenta/client/backend/types/environment_output.py,sha256=dl0GKodeqB7kWK5mH6Y4iBppkpwRzSTmtkXH1II4L6w,1257
@@ -73,7 +73,7 @@ agenta/client/backend/types/evaluation_webhook.py,sha256=emuKpD4NQjgYkjvQUyKLzHH
 agenta/client/backend/types/evaluator.py,sha256=AezEarRlPQ19mDU5Af4zixwbWckbMPg8Kjn11HxyXFQ,1084
 agenta/client/backend/types/evaluator_config.py,sha256=d3_ZxThl6ZkXTbmtBiqIg6CJIH8ha6am6fPEYcxp89A,1124
 agenta/client/backend/types/feedback.py,sha256=aYDUHvxJ0A3Dhs8JSWYLpxef3bIaHrug6gXH5pNtuAQ,1136
-agenta/client/backend/types/get_config_response.py,sha256=8a-RJDiBewC5Hdc0sZgt7sGwBfFzSohYqXIlQF1azf0,1081
+agenta/client/backend/types/get_config_response.py,sha256=iMxQvS9TiUy5JZdQF4Zeh9GroJLME17if1bsrWtwTEs,1045
 agenta/client/backend/types/http_validation_error.py,sha256=YhAPLKvuWLcgHn-uVWiZdWInseMiIS5eymdHKDqAB6Y,1060
 agenta/client/backend/types/human_evaluation.py,sha256=fSSO8Li0pAPcYe6lVUvJfrBTeGlJ8srdSw478vxXwW0,1305
 agenta/client/backend/types/human_evaluation_scenario.py,sha256=Ug3XQusHX5VfU_X2_EJhU58xgedZi59MrRIAgJWk3VY,1508
@@ -128,12 +128,12 @@ agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6F
 agenta/docker/docker_utils.py,sha256=ueA-t4-UCZDemEMPUX6GqST-0Ewfc05G3GKkmRJuK4w,3422
 agenta/sdk/__init__.py,sha256=jmeLRuXrew02ZruODZYIu4kpw0S8vV6JhMPQWFGtj30,648
 agenta/sdk/agenta_decorator.py,sha256=aYTsTGd22X87CxyqGIyfNGcNjVZCaO8yDk_M7US9Dpw,17287
-agenta/sdk/agenta_init.py,sha256=KHiIBtItXxOAkVtuFQfwokvw1_Xop8cCKYoVoqCMx5g,8886
+agenta/sdk/agenta_init.py,sha256=wDfStpe8_3ZXRLtikarwDKI_VpA1YW4eIz_3fXq39is,9044
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
 agenta/sdk/tracing/decorators.py,sha256=ujtU8gf3GDoHYuLTfEYK_2eIYZ-1oX5dpv02Mf4l_II,1191
-agenta/sdk/tracing/llm_tracing.py,sha256=kJ1yrl6quoswIgxR-dBQBAdmUDhWU_X4K4ByNS52J1k,7175
+agenta/sdk/tracing/llm_tracing.py,sha256=UiotJ56EFA3VPt7LREkcK2w51D9-0T1QNvBy4zNWEdY,7348
 agenta/sdk/tracing/logger.py,sha256=4zG9c51p8xPdKA5SL8MOgBfkpCnBSuV6JfWiXO0A7oc,473
 agenta/sdk/tracing/tasks_manager.py,sha256=XVGBEOwmHa6KcCC0PApk0_bZ0Ilk2ESuduNObB1rw2s,3792
 agenta/sdk/types.py,sha256=Mn0yBlHh_Yr_5oQXUfsYI3V7sJAVWkJgkxEOBDOOMS0,5852
@@ -155,7 +155,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.13.0a1.dist-info/METADATA,sha256=opQ9aTTLZU5okKXpy-RmQRre36x9D8S_e-p4rolfUuw,26433
-agenta-0.13.0a1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.13.0a1.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.13.0a1.dist-info/RECORD,,
+agenta-0.13.0a2.dist-info/METADATA,sha256=SplganFRsle75uYd8G-BFLHPXjuajj_iyhtXNIE3mkg,26433
+agenta-0.13.0a2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.13.0a2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.13.0a2.dist-info/RECORD,,