agenta 0.13.0a1__tar.gz → 0.13.0a3__tar.gz

{agenta-0.13.0a1 → agenta-0.13.0a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.13.0a1
+Version: 0.13.0a3
 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 → agenta-0.13.0a3}/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-0.13.0a1 → agenta-0.13.0a3}/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-0.13.0a1 → agenta-0.13.0a3}/agenta/sdk/agenta_decorator.py

@@ -121,7 +121,7 @@ def entrypoint(func: Callable[..., Any]) -> Callable[..., Any]:
             name=func.__name__,
             inputs=func_params,
             config=config,
-            environment="playground",  # type: ignore #NOTE: wrapper is only called in playground
+            environment=kwargs["environment"],  # type: ignore #NOTE: wrapper is only called in playground
         )
 
         llm_result = await execute_function(

{agenta-0.13.0a1 → agenta-0.13.0a3}/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-0.13.0a1 → agenta-0.13.0a3}/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 → agenta-0.13.0a3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.13.0a1"
+version = "0.13.0a3"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]