agenta 0.27.0a13__tar.gz → 0.27.1__tar.gz

{agenta-0.27.0a13 → agenta-0.27.1}/PKG-INFO

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

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/__init__.py

@@ -14,6 +14,7 @@ from .sdk.types import (
     TextParam,
     FileInputURL,
     BinaryParam,
+    Prompt,
 )
 
 from .sdk.utils.logging import log as logging

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/sdk/__init__.py

@@ -16,6 +16,7 @@ from .types import (
     MessagesInput,
     FileInputURL,
     BinaryParam,
+    Prompt,
 )
 
 from .tracing import Tracing, get_tracer

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/sdk/agenta_init.py

@@ -79,13 +79,13 @@ class AgentaSingleton:
         )
 
         self.app_id = app_id or config.get("app_id") or os.environ.get("AGENTA_APP_ID")
-        if not self.app_id:
-            raise ValueError(
-                "App ID must be specified. You can provide it in one of the following ways:\n"
-                "1. As an argument when calling ag.init(app_id='your_app_id').\n"
-                "2. In the configuration file specified by config_fname.\n"
-                "3. As an environment variable 'AGENTA_APP_ID'."
-            )
+        # if not self.app_id:
+        #     raise ValueError(
+        #         "App ID must be specified. You can provide it in one of the following ways:\n"
+        #         "1. As an argument when calling ag.init(app_id='your_app_id').\n"
+        #         "2. In the configuration file specified by config_fname.\n"
+        #         "3. As an environment variable 'AGENTA_APP_ID'."
+        #     )
 
         self.api_key = (
             api_key or os.environ.get("AGENTA_API_KEY") or config.get("api_key")

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/sdk/managers/config.py

@@ -93,7 +93,7 @@ class ConfigManager:
         return parameters
 
     @staticmethod
-    async def async_get_from_route(
+    async def aget_from_route(
         schema: Optional[Type[T]] = None,
     ) -> Union[Dict[str, Any], T]:
         """
@@ -214,7 +214,7 @@ class ConfigManager:
         return config.params
 
     @staticmethod
-    async def async_get_from_registry(
+    async def aget_from_registry(
         schema: Optional[Type[T]] = None,
         #
         app_id: Optional[str] = None,

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/sdk/tracing/inline.py

@@ -821,11 +821,15 @@ def parse_from_otel_span_dto(
 
     node_id = UUID(tree_id.hex[16:] + otel_span_dto.context.span_id[2:])
 
-    node_type: str = types.get("node")
+    node_type = NodeType.TASK
+    try:
+        node_type = NodeType(types.get("node", "").lower())
+    except:  # pylint: disable=bare-except
+        pass
 
     node = NodeDTO(
         id=node_id,
-        type=node_type.lower() if node_type else None,
+        type=node_type,
         name=otel_span_dto.name,
     )
 

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/sdk/tracing/tracing.py

@@ -35,8 +35,6 @@ class Tracing(metaclass=Singleton):
     ) -> None:
         # ENDPOINT (OTLP)
         self.otlp_url = url
-        # AITH (OTLP)
-        self.api_key: Optional[str] = None
         # HEADERS (OTLP)
         self.headers: Dict[str, str] = dict()
         # REFERENCES
@@ -59,16 +57,12 @@ class Tracing(metaclass=Singleton):
         # DEPRECATING
         app_id: Optional[str] = None,
     ):
-        # AUTH (OTLP)
-        self.api_key = api_key
         # HEADERS (OTLP)
-        self.headers = {}
-        if app_id:
-            self.headers.update(**{"AG-APP-ID": app_id})
         if api_key:
-            self.headers.update(**{"Authorization": self.api_key})
+            self.headers["Authorization"] = api_key
         # REFERENCES
-        self.references["application.id"] = app_id
+        if app_id:
+            self.references["application.id"] = app_id
 
         # TRACER PROVIDER
         self.tracer_provider = TracerProvider(

{agenta-0.27.0a13 → agenta-0.27.1}/agenta/sdk/types.py

@@ -222,7 +222,10 @@ class LifecyclesResponse(ReferencesResponse):
     deployed_by_id: Optional[str] = None
 
     def __str__(self):
-        return str(self.model_dump(exclude_none=True))
+        return self.model_dump_json(indent=4)
+
+    def __repr__(self):
+        return self.__str__()
 
 
 class ConfigurationResponse(LifecyclesResponse):
@@ -231,3 +234,14 @@ class ConfigurationResponse(LifecyclesResponse):
 
 class DeploymentResponse(LifecyclesResponse):
     pass
+
+
+class Prompt(BaseModel):
+    temperature: float
+    model: str
+    max_tokens: int
+    prompt_system: str
+    prompt_user: str
+    top_p: float
+    frequency_penalty: float
+    presence_penalty: float

{agenta-0.27.0a13 → agenta-0.27.1}/pyproject.toml

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