agenta 0.25.4a3__py3-none-any.whl → 0.25.4a4__py3-none-any.whl

agenta/sdk/tracing/tasks_manager.py

@@ -0,0 +1,129 @@
+# Stdlib Imports
+import queue
+import asyncio
+from logging import Logger
+from typing import Coroutine, Optional, Union
+from concurrent.futures import ThreadPoolExecutor
+
+# Own Imports
+from agenta.client.backend.types.error import Error
+from agenta.client.backend.client import AsyncObservabilityClient
+
+
+class AsyncTask(object):
+    """Wraps a coroutine (an asynchronous function defined with async def).
+
+    Args:
+        coroutine (Coroutine): asynchronous function
+    """
+
+    def __init__(
+        self,
+        coroutine_id: str,
+        coroutine_type: str,
+        coroutine: Coroutine,
+        client: AsyncObservabilityClient,
+    ):
+        self.coroutine_id = coroutine_id
+        self.coroutine_type = coroutine_type
+        self.coroutine = coroutine
+        self.task: Optional[asyncio.Task] = None
+        self.client = client
+
+    async def run(self) -> Union[asyncio.Task, Error]:
+        """Creates an asyncio Task from the coroutine and starts it
+
+        Returns:
+            Task: asyncio task
+        """
+
+        try:
+            self.task = asyncio.create_task(self.coroutine)
+        except Exception as exc:
+            return Error(message="error running task", stacktrace=str(exc))
+        return await self.task
+
+    def cancel(self):
+        """
+        Cancels running asyncio Task.
+        """
+
+        if self.task:
+            self.task.cancel()
+
+
+class TaskQueue(object):
+    """Stores a list of AsyncTask instances.
+
+    Args:
+        tasks (List[AsyncTasks]): list of async task instances
+
+    Example Usage:
+        ```python
+        queue = TaskQueue()
+        queue.add_task(long_running_task(1))
+        queue.add_task(long_running_task(2))
+        ```
+    """
+
+    def __init__(self, num_workers: int, logger: Logger):
+        self.tasks = queue.Queue()  # type: ignore
+        self._logger = logger
+        self._thread_pool = ThreadPoolExecutor(max_workers=num_workers)
+
+    def add_task(
+        self,
+        coroutine_id: str,
+        coroutine_type: str,
+        coroutine: Coroutine,
+        obs_client: AsyncObservabilityClient,
+    ) -> AsyncTask:
+        """Adds a new task to be executed.
+
+        Args:
+            coroutine_id (str): The Id of the coroutine
+            coroutine_type (str): The type of coroutine
+            coroutine (Coroutine): async task
+            obs_client (AsyncObservabilityClient): The async observability client
+
+        Returns:
+            AsyncTask: task to be executed
+        """
+
+        task = AsyncTask(coroutine_id, coroutine_type, coroutine, obs_client)
+        self.tasks.put(task)
+        return self._worker()
+
+    def _worker(self):
+        """
+        Runs task gotten from the queue in a thread pool.
+        """
+
+        while True:
+            task: AsyncTask = self.tasks.get()  # type: ignore
+            try:
+                future = self._thread_pool.submit(asyncio.run, task.run())
+                future.result()
+            except Exception as exc:
+                self._logger.error(
+                    f"Task '{task.coroutine_type}' failed with error: {str(exc)}"
+                )
+                break
+            finally:
+                self.tasks.task_done()
+                break
+
+    def _get_size(self) -> int:
+        """Returns the approximate number of items in the queue."""
+
+        return self.tasks.qsize()
+
+    def flush(self) -> None:
+        """Clears all items from the queue."""
+
+        q_size = self._get_size()
+        self._logger.info("Flushing queue...")
+        with self.tasks.mutex:  # acts as a lock to ensure that only one thread can access the queue
+            self.tasks.join()
+            self._logger.info(f"Queue with {q_size} items flushed successfully")
+            return

agenta/sdk/tracing/tracing_context.py

@@ -0,0 +1,27 @@
+from contextvars import ContextVar
+
+from typing import Optional, Dict, List
+
+from agenta.client.backend.types.create_span import CreateSpan
+
+CURRENT_TRACING_CONTEXT_KEY = "current_tracing_context"
+
+
+class TracingContext:
+    def __init__(self):
+        ### --- TRACE --- ###
+        self.trace_id: Optional[str] = None
+        self.trace_tags: List[str] = []
+
+        ### --- SPANS --- ###
+        self.active_span: Optional[CreateSpan] = None
+        self.spans: Dict[str, CreateSpan] = {}
+
+    def __repr__(self) -> str:
+        return f"TracingContext(trace='{self.trace_id}', spans={[f'{span.id} {span.spankind}' for span in self.spans.values()]})"
+
+    def __str__(self) -> str:
+        return self.__repr__()
+
+
+tracing_context = ContextVar(CURRENT_TRACING_CONTEXT_KEY, default=None)

agenta/sdk/types.py

@@ -186,3 +186,15 @@ class FileInputURL(HttpUrl):
     @classmethod
     def __schema_type_properties__(cls) -> dict:
         return {"x-parameter": "file_url", "type": "string"}
+
+
+class Context(BaseModel):
+    model_config = ConfigDict(extra="allow")
+
+    def to_json(self):
+        return self.model_dump()
+
+    @classmethod
+    def from_json(cls, json_str: str):
+        data = json.loads(json_str)
+        return cls(**data)

agenta/sdk/utils/debug.py

@@ -1,7 +1,7 @@
 from inspect import iscoroutinefunction
 from functools import wraps
 
-from agenta.sdk.utils.logging import log
+from agenta.sdk.tracing.logger import llm_logger as logging
 
 DEBUG = False
 SHIFT = 7
@@ -14,7 +14,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
         @wraps(f)
         async def async_log_wrapper(*args, **kwargs):
             if DEBUG:
-                log.debug(
+                logging.debug(
                     " ".join(
                         [
                             chars[0] * shift + " " * (SHIFT - shift),
@@ -26,7 +26,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
                 )
             result = await f(*args, **kwargs)
             if DEBUG:
-                log.debug(
+                logging.debug(
                     " ".join(
                         [
                             chars[1] * shift + " " * (SHIFT - shift),
@@ -40,7 +40,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
         @wraps(f)
         def log_wrapper(*args, **kwargs):
             if DEBUG:
-                log.debug(
+                logging.debug(
                     " ".join(
                         [
                             chars[0] * shift + " " * (SHIFT - shift),
@@ -52,7 +52,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
                 )
             result = f(*args, **kwargs)
             if DEBUG:
-                log.debug(
+                logging.debug(
                     " ".join(
                         [
                             chars[1] * shift + " " * (SHIFT - shift),

agenta/sdk/utils/globals.py

@@ -1,13 +1,15 @@
 import agenta
 
 
-def set_global(config=None, tracing=None):
-    """Allows usage of agenta.config and agenta.tracing in the user's code.
+def set_global(setup=None, config=None, tracing=None):
+    """Allows usage of agenta.config and agenta.setup in the user's code.
 
     Args:
+        setup: _description_. Defaults to None.
         config: _description_. Defaults to None.
-        tracing: _description_. Defaults to None.
     """
+    if setup is not None:
+        agenta.setup = setup
     if config is not None:
         agenta.config = config
     if tracing is not None:

agenta/sdk/utils/{costs.py → helper/openai_cost.py}

@@ -49,9 +49,6 @@ MODEL_COST_PER_1K_TOKENS = {
     "gpt-35-turbo-16k-completion": 0.004,
     "gpt-35-turbo-16k-0613-completion": 0.004,
     # Others
-    "text-embedding-ada-002": 0.1,
-    "text-ada-002": 0.1,
-    "adav2": 0.1,
     "text-ada-001": 0.0004,
     "ada": 0.0004,
     "text-babbage-001": 0.0005,

{agenta-0.25.4a3.dist-info → agenta-0.25.4a4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.25.4a3
+Version: 0.25.4a4
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -23,10 +23,6 @@ Requires-Dist: fastapi (>=0.100.0)
 Requires-Dist: httpx (>=0.24,<0.28)
 Requires-Dist: importlib-metadata (>=8.0.0,<9.0)
 Requires-Dist: ipdb (>=0.13)
-Requires-Dist: litellm (>=1.48.0,<2.0.0)
-Requires-Dist: opentelemetry-api (>=1.27.0,<2.0.0)
-Requires-Dist: opentelemetry-exporter-otlp (>=1.27.0,<2.0.0)
-Requires-Dist: opentelemetry-sdk (>=1.27.0,<2.0.0)
 Requires-Dist: posthog (>=3.1.0,<4.0.0)
 Requires-Dist: pydantic (>=2)
 Requires-Dist: pymongo (>=4.6.3,<5.0.0)

{agenta-0.25.4a3.dist-info → agenta-0.25.4a4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agenta/__init__.py,sha256=cd3c7bpyfZC8ho2Dc299bg8l_BDRQ3TFbd9Tnn2UQTY,923
+agenta/__init__.py,sha256=MCeFBzYtrVNdhElvl1fBjypsqhgEaAMKy6ZR-B3YmJk,993
 agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxugwd8,474
 agenta/cli/helper.py,sha256=vRxHyeNaltzNIGrfU2vO0H28_rXDzx9QqIZ_S-W6zL4,6212
 agenta/cli/main.py,sha256=Wz0ODhoeKK3Qg_CFUhu6D909szk05tc8ZVBB6H1-w7k,9763
@@ -14,7 +14,7 @@ agenta/client/backend/apps/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2t
 agenta/client/backend/apps/client.py,sha256=9rWax7LlEIpXdqq4McPAkvwHweSTomnyrFxFo8ehYy0,54136
 agenta/client/backend/bases/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/bases/client.py,sha256=BZsz5eXaa2voZdJXqgd5J5hPUuYvWwIcPCWyl49w-oY,6028
-agenta/client/backend/client.py,sha256=k262wVUWPYWbZhLiZec7sTLfh8CD7tfCYdFKyLqUMi0,102659
+agenta/client/backend/client.py,sha256=NqKTA7oKQFI4eU0TNfB-SEDILMgVEJSZFodGsl3hWZI,102435
 agenta/client/backend/configs/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/configs/client.py,sha256=qtrRdRuNIZBjh9Ykj7eNKhocp7dFdFQoDSbUduVjKJ0,19089
 agenta/client/backend/containers/__init__.py,sha256=Haw2PwiPhNvM26PLQN57jY0bN-QqPoDG4VA-P_uGL3A,153
@@ -26,9 +26,9 @@ agenta/client/backend/core/api_error.py,sha256=TtMtCdxXjd7Tasc9c8ooFg124nPrb2MXG
 agenta/client/backend/core/client_wrapper.py,sha256=mU22mZBjz2QXalKJLDGo1qNeAt1awLxRMO8fzbPbnr8,1829
 agenta/client/backend/core/datetime_utils.py,sha256=BHjt_H3WVslcuPsr6qjJoVif_SsdLvFN0c43ABE5UiQ,1069
 agenta/client/backend/core/file.py,sha256=vliNmlB7PbDfi4EKiVPNq5QaGXJ4zlDBGupv7Qciy7g,1520
-agenta/client/backend/core/http_client.py,sha256=1ORQxbKS0c6nMbeiANucMrlwUImmLXlCfjBbR_W-LSY,20440
+agenta/client/backend/core/http_client.py,sha256=LHzx7H12q9WOUquq8QZYSUPCt4TRdwxcRbwQm2JboeE,20278
 agenta/client/backend/core/jsonable_encoder.py,sha256=SHXw4G4n-f0IPgNkxj_-Fip3kN8NUAI-YrKxdZw8kl0,3662
-agenta/client/backend/core/pydantic_utilities.py,sha256=jKArwRgKxutDYiTxssKykAGFHQM6_m9G-hLFCXVB8yU,7617
+agenta/client/backend/core/pydantic_utilities.py,sha256=sLzQZQLS_y66SZSBMtJE7YrQK8INda-ta1tKwAHabGk,7540
 agenta/client/backend/core/query_encoder.py,sha256=8qYl5VPl1jU4cDF0X7oSU_DXjlVWY5ayigFBpNTMGOA,2150
 agenta/client/backend/core/remove_none_from_dict.py,sha256=EU9SGgYidWq7SexuJbNs4-PZ-5Bl3Vppd864mS6vQZw,342
 agenta/client/backend/core/request_options.py,sha256=5cCGt5AEGgtP5xifDl4oVQUmSjlIA8FmRItAlJawM18,1417
@@ -132,38 +132,28 @@ agenta/docker/docker-assets/entrypoint.sh,sha256=29XK8VQjQsx4hN2j-4JDy-6kQb5y4LC
 agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4JrYu9rLz3I-LxCfeEg,83
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=kO1q2_IR0fEAo4M-2Pt_v-zC7GxxnkLogjKFhU869Ps,3555
-agenta/sdk/__init__.py,sha256=k6lzJF6gzaFdzq_FJ3chBu6ug8pNCT3_-bfeB1ZqGkg,1165
-agenta/sdk/agenta_init.py,sha256=dyUV3vDI55XYQ3hNxkT7ibe-PSPIWz4eO4skTkjxOtk,10603
+agenta/sdk/__init__.py,sha256=XyxMUm8-fCH6G6KAFU0iAfiKTGEzNzBtbAfCp9i1Y7o,831
+agenta/sdk/agenta_init.py,sha256=8MfDuypxohd0qRTdtGjX7L17KW-1UGmzNVdiqF15_ak,9790
 agenta/sdk/assets.py,sha256=Zv4i8MVUSB3jMODQon1mzJtYxuntmrCNjLGk8f-2fls,2856
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
-agenta/sdk/config_manager.py,sha256=HFOJpJKBkhlA0C-KPuxb4-bHNZeZqdpmx_beoX4lQw8,7997
-agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/context/routing.py,sha256=gOoOM88hSjIjzQ3ni68TjUoq7WtwSH3kB07YjCYvW2c,631
-agenta/sdk/context/tracing.py,sha256=UmmW15UFFsvxS0myS6aD9wBk5iNepNlQi4tEQ_ejfYM,96
-agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=4vJsFwyXUx1cuzILVOzELDfv17QnYTXpIE2NwFnC__I,35427
-agenta/sdk/decorators/tracing.py,sha256=atMLiRpQCFSAVK-rPlN781DUq7gARe0XgJJHBKqBAUc,11657
-agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
-agenta/sdk/litellm/litellm.py,sha256=j4WyRTQdxLkvelaPm5BfUbaUxkbQXDhyr3yp9yXo1RQ,8289
-agenta/sdk/router.py,sha256=mOguvtOwl2wmyAgOuWTsf98pQwpNiUILKIo67W_hR3A,119
-agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD98,41
-agenta/sdk/tracing/attributes.py,sha256=0Ld0A4eL6EAWwbhBY0Qp7MLWXXbBsd8EDtbBuwIy1PI,4775
-agenta/sdk/tracing/context.py,sha256=KDQYjRcmbpb6e0xEuX-bAQGhEjwx4H2nwybRPOKCDjo,578
-agenta/sdk/tracing/conventions.py,sha256=YFkKPls77sl5YOXrRjfn4erEjczMXThc7b-rRaVVTuo,722
-agenta/sdk/tracing/exporters.py,sha256=e-SfpN8dKQ0vuBZQTZ6qgIMBhyriOrVeLYJkv-us0Zs,1328
-agenta/sdk/tracing/inline.py,sha256=o-lRHmcIXVrFLWIj-xeY9LNbkuR8G4IUubwlWVfNC30,35769
-agenta/sdk/tracing/processors.py,sha256=PRQpj8aOmOjvf71M48AtCurIxVFg2ZZWRtJLtlB4xyE,1974
-agenta/sdk/tracing/spans.py,sha256=AnQuuyuCGST6n_He4SF5WdrtMts9JlE8v6rVRdm2rRg,3570
-agenta/sdk/tracing/tracing.py,sha256=5r0whzLzF2bwYKZkUN3QDjq8AketEb-9sM4nqnO9zzM,5168
-agenta/sdk/types.py,sha256=B7C7Jpeq0CpmLwmN1dmpggJsZX8WpxyyM42PkdSsx3s,5669
+agenta/sdk/config_manager.py,sha256=n3UTpIhiB1J6FIQ5IvlucfGVpxoIIemnmdI68p34Yvw,8003
+agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
+agenta/sdk/decorators/base.py,sha256=9aNdX5h8a2mFweuhdO-BQPwXGKY9ONPIdLRhSGAGMfY,217
+agenta/sdk/decorators/llm_entrypoint.py,sha256=jXXTQS-sCwmkKvF0bNYTG9pkK13B2odnHsrYepTU_x8,35572
+agenta/sdk/decorators/tracing.py,sha256=e0olx2EEdjXY0NqpIoDJSVxCnUmv0woewTUuCJXy2tM,4166
+agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
+agenta/sdk/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agenta/sdk/tracing/callbacks.py,sha256=ug9xeXLgJEpx4l3x1BrJD2bFrH83I3kajlGsU8WtG4U,7699
+agenta/sdk/tracing/llm_tracing.py,sha256=5EeIuRu2wgGknlsZD-AkuJH_gOBrmsrCrWK0-h0Uov0,17941
+agenta/sdk/tracing/logger.py,sha256=GfH7V-jBHcn7h5dbdrnkDMe_ml3wkXFBeoQiqR4KVRc,474
+agenta/sdk/tracing/tasks_manager.py,sha256=FBSFOWIKBycyA4ShB2ZVMzrzYQ8pWGWWBClFX8nlZFA,3726
+agenta/sdk/tracing/tracing_context.py,sha256=nt3ewa-TK9BRJviGIZYazsAQUiG4daWxjtsbjeaDprs,789
+agenta/sdk/types.py,sha256=1AXUKTURFFgPH5iWeTYcNcg6dxVo1RrD0WuVe9e-NgU,5919
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/utils/costs.py,sha256=i8C7ud__pThLS55XkN4YW8czXtGeXr2mx7jjcOFeiXg,5955
-agenta/sdk/utils/debug.py,sha256=DxiCAeXxxrcEZT2CjlNA6BMvujGP4nzQ-rfb-_mLMck,2114
-agenta/sdk/utils/exceptions.py,sha256=kzU5bNBMZvVv0xnwvc5osVMsb_Ujm0qqeem9wpNo2z0,500
-agenta/sdk/utils/globals.py,sha256=2HhyzWn55BbYNCZ3rT8dAxk1GGXuGQPbtq_THjaHbBw,372
-agenta/sdk/utils/logging.py,sha256=FKzAO5eHjR_qvpEnc4tKXUA6WftmwdwYwuJK6CbSc84,379
+agenta/sdk/utils/debug.py,sha256=QyuPsSoN0425UD13x_msPxSF_VT6YwHiQunZUibI-jg,2149
+agenta/sdk/utils/globals.py,sha256=JmhJcCOSbwvjQ6GDyUc2_SYR27DZk7YcrRH80ktHHOM,435
+agenta/sdk/utils/helper/openai_cost.py,sha256=1VkgvucDnNZm1pTfcVLz9icWunntp1d7zwMmnviy3Uw,5877
 agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
-agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/templates/compose_email/README.md,sha256=ss7vZPpI1Hg0VmYtFliwq_r5LnqbCy_S5OQDXg8UoIA,308
 agenta/templates/compose_email/app.py,sha256=SG8QnBkC49MtvWa2z5_YQ2d8wxZxhO_hWs15j3CsWvc,2384
 agenta/templates/compose_email/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
@@ -179,7 +169,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.25.4a3.dist-info/METADATA,sha256=Ve2s3ouTz56d41dEWduIVk06vpAUxFxvq_lqU_0OLSs,31738
-agenta-0.25.4a3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-agenta-0.25.4a3.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.25.4a3.dist-info/RECORD,,
+agenta-0.25.4a4.dist-info/METADATA,sha256=XSV7qwA3Ym03XcjA8Z-qNNZnnWPiERd2g5okOzxgHUM,31534
+agenta-0.25.4a4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+agenta-0.25.4a4.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.25.4a4.dist-info/RECORD,,

agenta/sdk/context/routing.py

@@ -1,25 +0,0 @@
-from contextlib import contextmanager
-from contextvars import ContextVar
-from typing import Any, Dict, Optional
-
-routing_context = ContextVar("routing_context", default={})
-
-
-@contextmanager
-def routing_context_manager(
-    config: Optional[Dict[str, Any]] = None,
-    environment: Optional[str] = None,
-    version: Optional[str] = None,
-    variant: Optional[str] = None,
-):
-    context = {
-        "config": config,
-        "environment": environment,
-        "version": version,
-        "variant": variant,
-    }
-    token = routing_context.set(context)
-    try:
-        yield
-    finally:
-        routing_context.reset(token)

agenta/sdk/context/tracing.py

@@ -1,3 +0,0 @@
-from contextvars import ContextVar
-
-tracing_context = ContextVar("tracing_context", default={})

agenta/sdk/litellm/__init__.py

@@ -1 +0,0 @@
-from .litellm import litellm_handler