ragaai-catalyst 1.0.8b4__py3-none-any.whl → 1.0.8.2__py3-none-any.whl

ragaai_catalyst/ragaai_catalyst.py

@@ -54,16 +54,21 @@ class RagaAICatalyst:
 
         self.api_keys = api_keys or {}
 
-        if self.api_keys:
-            self._upload_keys()
-
         if base_url:
             RagaAICatalyst.BASE_URL = base_url
             try:
-                self.get_token()
                 os.environ["RAGAAI_CATALYST_BASE_URL"] = base_url
             except requests.exceptions.RequestException:
-                raise ConnectionError("The provided base_url is not accessible. Please re-check the base_url.")
+                raise ConnectionError(
+                    "The provided base_url is not accessible. Please re-check the base_url."
+                )
+
+        # Get the token from the server
+        self.get_token()
+
+        # Set the API keys, if  available
+        if self.api_keys:
+            self._upload_keys()
 
     def _set_access_key_secret_key(self, access_key, secret_key):
         os.environ["RAGAAI_CATALYST_ACCESS_KEY"] = access_key
@@ -145,10 +150,7 @@ class RagaAICatalyst:
             return None
 
         headers = {"Content-Type": "application/json"}
-        json_data = {
-            "accessKey": access_key,
-            "secretKey": secret_key
-        }
+        json_data = {"accessKey": access_key, "secretKey": secret_key}
 
         response = requests.post(
             f"{ RagaAICatalyst.BASE_URL}/token",
@@ -161,7 +163,9 @@ class RagaAICatalyst:
         if response.status_code == 400:
             token_response = response.json()
             if token_response.get("message") == "Please enter valid credentials":
-                raise Exception("Authentication failed. Invalid credentials provided. Please check your Access key and Secret key. \nTo view or create new keys, navigate to Settings -> Authenticate in the RagaAI Catalyst dashboard.")
+                raise Exception(
+                    "Authentication failed. Invalid credentials provided. Please check your Access key and Secret key. \nTo view or create new keys, navigate to Settings -> Authenticate in the RagaAI Catalyst dashboard."
+                )
 
         response.raise_for_status()
 
@@ -183,7 +187,6 @@ class RagaAICatalyst:
             logger.error("Token(s) not set")
             return None
 
-
     def create_project(self, project_name, type="llm", description=""):
         """
         Creates a project with the given project_name, type, and description.
@@ -253,7 +256,6 @@ class RagaAICatalyst:
                 "Unexpected error while creating project: %s", str(general_err1)
             )
             return "An unexpected error occurred while creating the project"
-        
 
     def list_projects(self, num_projects=100):
         """

{ragaai_catalyst-1.0.8b4.dist-info → ragaai_catalyst-1.0.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 1.0.8b4
+Version: 1.0.8.2
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>, Siddhartha Kosti <siddhartha.kosti@raga.ai>, Ritika Goel <ritika.goel@raga.ai>, Vijay Chaurasia <vijay.chaurasia@raga.ai>
 Requires-Python: >=3.9

{ragaai_catalyst-1.0.8b4.dist-info → ragaai_catalyst-1.0.8.2.dist-info}/RECORD

@@ -3,7 +3,7 @@ ragaai_catalyst/_version.py,sha256=JKt9KaVNOMVeGs8ojO6LvIZr7ZkMzNN-gCcvryy4x8E,4
 ragaai_catalyst/dataset.py,sha256=xEun-MqTgT0JGyGB2t0BwwC6M7t1aUT7x5X12C-_4pI,7644
 ragaai_catalyst/experiment.py,sha256=c0l72fNULb4KZVcXwehZKdXb116JRyd0Y3hU3CVf8Fs,18999
 ragaai_catalyst/prompt_manager.py,sha256=3jEcgS0JRZzjHww_ityOkq9SMvi7DQ75PlKu-17dXW4,14729
-ragaai_catalyst/ragaai_catalyst.py,sha256=LJECRdCmW8DtF3Lk4wZ6lXJgsYkh6Oome45DzONyaVU,16228
+ragaai_catalyst/ragaai_catalyst.py,sha256=hTkGU9-05TIRuePJ5I0-FKhr_WzlNtGILPCzHKs9tZw,16335
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/tracers/__init__.py,sha256=NppmJhD3sQ5R1q6teaZLS7rULj08Gb6JT8XiPRIe_B0,49
 ragaai_catalyst/tracers/tracer.py,sha256=LZQNKQF6hRfzxDu_Ljs2HBxQinjihKQoH189gIsO4w4,9171
@@ -16,7 +16,7 @@ ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hak
 ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-1.0.8b4.dist-info/METADATA,sha256=tcv_Dor48L1tOy9zVmkkXU1juvkWNzTSe7tmMBFLnCs,6499
-ragaai_catalyst-1.0.8b4.dist-info/WHEEL,sha256=5Mi1sN9lKoFv_gxcPtisEVrJZihrm_beibeg5R6xb4I,91
-ragaai_catalyst-1.0.8b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-1.0.8b4.dist-info/RECORD,,
+ragaai_catalyst-1.0.8.2.dist-info/METADATA,sha256=GvMbIUGcMNHVAJa7VAX6GQFicnnT55_0xB906EsdkE0,6499
+ragaai_catalyst-1.0.8.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ragaai_catalyst-1.0.8.2.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-1.0.8.2.dist-info/RECORD,,

{ragaai_catalyst-1.0.8b4.dist-info → ragaai_catalyst-1.0.8.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.0.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any