agenta 0.48.10__py3-none-any.whl → 0.49.0__py3-none-any.whl

agenta/sdk/litellm/mocks/__init__.py

@@ -3,6 +3,8 @@ from asyncio import sleep
 
 from pydantic import BaseModel
 
+import agenta as ag
+
 
 class MockMessageModel(BaseModel):
     content: str
@@ -16,6 +18,7 @@ class MockResponseModel(BaseModel):
     choices: list[MockChoiceModel]
 
 
+@ag.instrument()
 def hello_mock_response(*args, **kwargs) -> MockResponseModel:
     return MockResponseModel(
         choices=[
@@ -28,6 +31,7 @@ def hello_mock_response(*args, **kwargs) -> MockResponseModel:
     )
 
 
+@ag.instrument()
 def chat_mock_response(*args, **kwargs) -> MockResponseModel:
     return MockResponseModel(
         choices=[
@@ -41,6 +45,7 @@ def chat_mock_response(*args, **kwargs) -> MockResponseModel:
     )
 
 
+@ag.instrument()
 def delay_mock_response(*args, **kwargs) -> MockResponseModel:
     sleep(2)
 
@@ -55,8 +60,147 @@ def delay_mock_response(*args, **kwargs) -> MockResponseModel:
     )
 
 
+@ag.instrument()
+def capital_mock_response(*args, **kwargs) -> MockResponseModel:
+    country = kwargs.get("messages", [{}, {}])[1].get(
+        "content", "What is the capital of _____?"
+    )[len("What is the capital of ") :][:-1]
+
+    capital = CAPITALS.get(country, "Unknown")
+    return MockResponseModel(
+        choices=[
+            MockChoiceModel(
+                message=MockMessageModel(
+                    content=f"The capital of {country} is {capital}.",
+                )
+            )
+        ],
+    )
+
+
 MOCKS: dict[str, Callable[..., MockResponseModel]] = {
     "hello": hello_mock_response,
     "chat": chat_mock_response,
     "delay": delay_mock_response,
+    "capital": capital_mock_response,
+}
+
+CAPITALS = {
+    "Afghanistan": "Kabul",
+    "Albania": "Tirana",
+    "Algeria": "Algiers",
+    "Andorra": "Andorra la Vella",
+    "Angola": "Luanda",
+    "Antigua and Barbuda": "St. John's",
+    "Argentina": "Buenos Aires",
+    "Armenia": "Yerevan",
+    "Australia": "Canberra",
+    "Austria": "Vienna",
+    "Azerbaijan": "Baku",
+    "Bahamas": "Nassau",
+    "Bahrain": "Manama",
+    "Bangladesh": "Dhaka",
+    "Barbados": "Bridgetown",
+    "Belarus": "Minsk",
+    "Belgium": "Brussels",
+    "Belize": "Belmopan",
+    "Benin": "Porto-Novo",
+    "Bhutan": "Thimphu",
+    "Bolivia": "Sucre",
+    "Bosnia and Herzegovina": "Sarajevo",
+    "Botswana": "Gaborone",
+    "Brazil": "Bras\u00edlia",
+    "Brunei": "Bandar Seri Begawan",
+    "Bulgaria": "Sofia",
+    "Burkina Faso": "Ouagadougou",
+    "Burundi": "Gitega",
+    "Cabo Verde": "Praia",
+    "Cambodia": "Phnom Penh",
+    "Cameroon": "Yaound\u00e9",
+    "Canada": "Ottawa",
+    "Central African Republic": "Bangui",
+    "Chad": "N'Djamena",
+    "Chile": "Santiago",
+    "China": "Beijing",
+    "Colombia": "Bogot\u00e1",
+    "Comoros": "Moroni",
+    "Congo (Brazzaville)": "Brazzaville",
+    "Congo (Kinshasa)": "Kinshasa",
+    "Costa Rica": "San Jos\u00e9",
+    "Croatia": "Zagreb",
+    "Cuba": "Havana",
+    "Cyprus": "Nicosia",
+    "Czech Republic": "Prague",
+    "Denmark": "Copenhagen",
+    "Djibouti": "Djibouti",
+    "Dominica": "Roseau",
+    "Dominican Republic": "Santo Domingo",
+    "East Timor": "Dili",
+    "Ecuador": "Quito",
+    "Egypt": "Cairo",
+    "El Salvador": "San Salvador",
+    "Equatorial Guinea": "Malabo",
+    "Eritrea": "Asmara",
+    "Estonia": "Tallinn",
+    "Eswatini": "Mbabane",
+    "Ethiopia": "Addis Ababa",
+    "Fiji": "Suva",
+    "Finland": "Helsinki",
+    "France": "Paris",
+    "Gabon": "Libreville",
+    "Gambia": "Banjul",
+    "Georgia": "Tbilisi",
+    "Germany": "Berlin",
+    "Ghana": "Accra",
+    "Greece": "Athens",
+    "Grenada": "St. George's",
+    "Guatemala": "Guatemala City",
+    "Guinea": "Conakry",
+    "Guinea-Bissau": "Bissau",
+    "Guyana": "Georgetown",
+    "Haiti": "Port-au-Prince",
+    "Honduras": "Tegucigalpa",
+    "Hungary": "Budapest",
+    "Iceland": "Reykjavik",
+    "India": "New Delhi",
+    "Indonesia": "Jakarta",
+    "Iran": "Tehran",
+    "Iraq": "Baghdad",
+    "Ireland": "Dublin",
+    "Israel": "Jerusalem",
+    "Italy": "Rome",
+    "Ivory Coast": "Yamoussoukro",
+    "Jamaica": "Kingston",
+    "Japan": "Tokyo",
+    "Jordan": "Amman",
+    "Kazakhstan": "Astana",
+    "Kenya": "Nairobi",
+    "Kiribati": "South Tarawa",
+    "North Korea": "Pyongyang",
+    "South Korea": "Seoul",
+    "Kosovo": "Pristina",
+    "Kuwait": "Kuwait City",
+    "Kyrgyzstan": "Bishkek",
+    "Laos": "Vientiane",
+    "Latvia": "Riga",
+    "Lebanon": "Beirut",
+    "Lesotho": "Maseru",
+    "Liberia": "Monrovia",
+    "Libya": "Tripoli",
+    "Liechtenstein": "Vaduz",
+    "Lithuania": "Vilnius",
+    "Luxembourg": "Luxembourg",
+    "Madagascar": "Antananarivo",
+    "Malawi": "Lilongwe",
+    "Malaysia": "Kuala Lumpur",
+    "Maldives": "Mal\u00e9",
+    "Mali": "Bamako",
+    "Malta": "Valletta",
+    "Marshall Islands": "Majuro",
+    "Mauritania": "Nouakchott",
+    "Mauritius": "Port Louis",
+    "Mexico": "Mexico City",
+    "Micronesia": "Palikir",
+    "Moldova": "Chi\u0219in\u0103u",
+    "Monaco": "Monaco",
 }

agenta/sdk/middleware/auth.py

@@ -100,8 +100,8 @@ class AuthMiddleware(BaseHTTPMiddleware):
             access_token = request.cookies.get("sAccessToken", None)
             cookies = {"sAccessToken": access_token} if access_token else None
 
-            if not headers and not cookies:
-                log.debug("No auth header nor auth cookie found in the request")
+            # if not headers and not cookies:
+            #     log.debug("No auth header nor auth cookie found in the request")
 
             # PARAMS
             params = {}
@@ -112,8 +112,9 @@ class AuthMiddleware(BaseHTTPMiddleware):
                 # ALTERNATIVE
                 or request.query_params.get("project_id")
             )
-            if not project_id:
-                log.debug("No project ID found in request")
+            # if not project_id:
+            #     log.debug("No project ID found in request")
+
             if project_id:
                 params["project_id"] = project_id
             ## SCOPE
@@ -139,7 +140,7 @@ class AuthMiddleware(BaseHTTPMiddleware):
                 credentials = _cache.get(_hash)
 
                 if credentials:
-                    log.debug("Using cached credentials")
+                    # log.debug("Using cached credentials")
                     return credentials
 
             try:
@@ -153,46 +154,46 @@ class AuthMiddleware(BaseHTTPMiddleware):
                             timeout=30.0,
                         )
                     except httpx.TimeoutException as exc:
-                        log.debug(f"Timeout error while verify credentials: {exc}")
+                        # log.debug(f"Timeout error while verify credentials: {exc}")
                         raise DenyException(
                             status_code=504,
                             content="Could not verify credentials: connection to {self.host} timed out. Please check your network connection.",
                         ) from exc
                     except httpx.ConnectError as exc:
-                        log.debug(f"Connection error while verify credentials: {exc}")
+                        # log.debug(f"Connection error while verify credentials: {exc}")
                         raise DenyException(
                             status_code=503,
                             content=f"Could not verify credentials: connection to {self.host} failed. Please check if agenta is available.",
                         ) from exc
                     except httpx.NetworkError as exc:
-                        log.debug(f"Network error while verify credentials: {exc}")
+                        # log.debug(f"Network error while verify credentials: {exc}")
                         raise DenyException(
                             status_code=503,
                             content="Could not verify credentials: connection to {self.host} failed. Please check your network connection.",
                         ) from exc
                     except httpx.HTTPError as exc:
-                        log.debug(f"HTTP error while verify credentials: {exc}")
+                        # log.debug(f"HTTP error while verify credentials: {exc}")
                         raise DenyException(
                             status_code=502,
                             content=f"Could not verify credentials: connection to {self.host} failed. Please check if agenta is available.",
                         ) from exc
 
                     if response.status_code == 401:
-                        log.debug("Agenta returned 401 - Invalid credentials")
+                        # log.debug("Agenta returned 401 - Invalid credentials")
                         raise DenyException(
                             status_code=401,
                             content="Invalid credentials. Please check your credentials or login again.",
                         )
                     elif response.status_code == 403:
-                        log.debug("Agenta returned 403 - Permission denied")
+                        # log.debug("Agenta returned 403 - Permission denied")
                         raise DenyException(
                             status_code=403,
                             content="Permission denied. Please check your permissions or contact your administrator.",
                         )
                     elif response.status_code != 200:
-                        log.debug(
-                            f"Agenta returned {response.status_code} - Unexpected status code"
-                        )
+                        # log.debug(
+                        #     f"Agenta returned {response.status_code} - Unexpected status code"
+                        # )
                         raise DenyException(
                             status_code=500,
                             content=f"Could not verify credentials: {self.host} returned unexpected status code {response.status_code}. Please try again later or contact support if the issue persists.",
@@ -201,16 +202,16 @@ class AuthMiddleware(BaseHTTPMiddleware):
                     try:
                         auth = response.json()
                     except ValueError as exc:
-                        log.debug(f"Agenta returned invalid JSON response: {exc}")
+                        # log.debug(f"Agenta returned invalid JSON response: {exc}")
                         raise DenyException(
                             status_code=500,
                             content=f"Could not verify credentials: {self.host} returned unexpected invalid JSON response. Please try again later or contact support if the issue persists.",
                         ) from exc
 
                     if not isinstance(auth, dict):
-                        log.debug(
-                            f"Agenta returned invalid response format: {type(auth)}"
-                        )
+                        # log.debug(
+                        #     f"Agenta returned invalid response format: {type(auth)}"
+                        # )
                         raise DenyException(
                             status_code=500,
                             content=f"Could not verify credentials: {self.host} returned unexpected invalid response format. Please try again later or contact support if the issue persists.",
@@ -218,7 +219,7 @@ class AuthMiddleware(BaseHTTPMiddleware):
 
                     effect = auth.get("effect")
                     if effect != "allow":
-                        log.debug("Access denied by Agenta - effect: {effect}")
+                        # log.debug("Access denied by Agenta - effect: {effect}")
                         raise DenyException(
                             status_code=403,
                             content="Permission denied. Please check your permissions or contact your administrator.",
@@ -226,8 +227,8 @@ class AuthMiddleware(BaseHTTPMiddleware):
 
                     credentials = auth.get("credentials")
 
-                    if not credentials:
-                        log.debug("No credentials found in the response")
+                    # if not credentials:
+                    #     log.debug("No credentials found in the response")
 
                     _cache.put(_hash, credentials)
 
@@ -236,9 +237,9 @@ class AuthMiddleware(BaseHTTPMiddleware):
             except DenyException as deny:
                 raise deny
             except Exception as exc:  # pylint: disable=bare-except
-                log.debug(
-                    f"Unexpected error while verifying credentials (remote): {exc}"
-                )
+                # log.debug(
+                #     f"Unexpected error while verifying credentials (remote): {exc}"
+                # )
                 raise DenyException(
                     status_code=500,
                     content=f"Could not verify credentials: unexpected error - {str(exc)}. Please try again later or contact support if the issue persists.",
@@ -247,7 +248,7 @@ class AuthMiddleware(BaseHTTPMiddleware):
         except DenyException as deny:
             raise deny
         except Exception as exc:
-            log.debug(f"Unexpected error while verifying credentials (local): {exc}")
+            # log.debug(f"Unexpected error while verifying credentials (local): {exc}")
             raise DenyException(
                 status_code=500,
                 content=f"Could not verify credentials: unexpected error - {str(exc)}. Please try again later or contact support if the issue persists.",

{agenta-0.48.10.dist-info → agenta-0.49.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agenta
-Version: 0.48.10
+Version: 0.49.0
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Keywords: LLMOps,LLM,evaluation,prompt engineering
 Author: Mahmoud Mabrouk

{agenta-0.48.10.dist-info → agenta-0.49.0.dist-info}/RECORD

@@ -318,7 +318,7 @@ agenta/sdk/decorators/tracing.py,sha256=6xwN9AomDND_5wL21NXFoZsdwTBuITw2I39GNqNG
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 agenta/sdk/litellm/litellm.py,sha256=dCjw0H_jD3L3UQ3l9SbLm5dfeIGnel6dPtQYJ78beYM,10202
 agenta/sdk/litellm/mockllm.py,sha256=PG3DF2ZjWHOtCt8WIU3qcSyXBOZspUX1FIyFMA9HuZM,902
-agenta/sdk/litellm/mocks/__init__.py,sha256=BH92MtQNPLfbN6ERk13XuDjVnNHy9joi7E_I3heUDC4,1299
+agenta/sdk/litellm/mocks/__init__.py,sha256=d-aAj-uzee1BVGHMB2oJocpxGQeWr6QLrYV4oNdSqzY,5241
 agenta/sdk/managers/__init__.py,sha256=SN-LRwG0pRRDV3u2Q4JiiSTigN3-mYpzGNM35RzT4mc,238
 agenta/sdk/managers/apps.py,sha256=BeAlTJlOOM0Wh_XWgjve65Mt-LwFgib_swu-1wvGQi4,2250
 agenta/sdk/managers/config.py,sha256=Dl1L4KGvj3h_temzJzn7FVHacABRuB2Q7_x98EpZJQo,7494
@@ -328,7 +328,7 @@ agenta/sdk/managers/shared.py,sha256=kI3w74E4rS3YDORG2e-_r0Pz5KslTRzRBK7vgyOeaJA
 agenta/sdk/managers/variant.py,sha256=A5ga3mq3b0weUTXa9HO72MGaspthGcu1uK9K5OnP738,4172
 agenta/sdk/managers/vault.py,sha256=054ce9X_xKa2M4NtQWz-GugO6q_pYVWCP3IxbAJJcRw,337
 agenta/sdk/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/middleware/auth.py,sha256=Iw7v7wsjHbVqjWc__hlwO6hMssd1azEl_uaE4H9FO18,10176
+agenta/sdk/middleware/auth.py,sha256=m7Ga6JpQ5CBCNA7nZYPw8NgS4Nihzkg-ptmvlLCV8TI,10227
 agenta/sdk/middleware/config.py,sha256=zd57kxaeYheZMG8nD5VOYgHXw8j9zRlFacuzCGHWeEA,7978
 agenta/sdk/middleware/cors.py,sha256=q3r7lGkrIdMcT_vuhsburMcjG7pyl7w0ycxrIrGJ2e8,921
 agenta/sdk/middleware/inline.py,sha256=ee8E4XBGcRSrHTvblqX1yRXuTN_sxLm7lY1jnywrBG8,901
@@ -357,6 +357,6 @@ agenta/sdk/utils/logging.py,sha256=j4NzpFk2ilOM10sRBOxCjmansDHSx6HwMV8IAEreRb8,8
 agenta/sdk/utils/preinit.py,sha256=1TAAHhYyYnLLwvzwnf33Qwkou7tI3iITlVt-1kcyGaM,1186
 agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/sdk/utils/timing.py,sha256=nZR-kudVUtKFlHuBhztgSGxj7FVnCB4Uv6sfg-1dkrQ,1556
-agenta-0.48.10.dist-info/METADATA,sha256=2RZE7x-3c5TdCAmvbyZzBnPRrpTuX7p6AUWLlLFZz9E,31457
-agenta-0.48.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-agenta-0.48.10.dist-info/RECORD,,
+agenta-0.49.0.dist-info/METADATA,sha256=i7oQJ0JGViTh2jXxvjsaQRJEd1IywFvDFvC7TBmQ3Kw,31456
+agenta-0.49.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+agenta-0.49.0.dist-info/RECORD,,