ragaai-catalyst 2.1.5b39__tar.gz → 2.1.5b41__tar.gz

{ragaai_catalyst-2.1.5b39/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.5b41}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b39
+Version: 2.1.5b41
 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>, Tushar Kumar <tushar.kumar@raga.ai>
-Requires-Python: <3.13,>=3.10
+Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/examples/langgraph_examples/customer_support.py

@@ -1,3 +1,6 @@
+import sys 
+sys.path.append('.')
+
 import os
 import random
 from dotenv import load_dotenv
@@ -11,14 +14,14 @@ from dotenv import load_dotenv
 load_dotenv()
 
 catalyst = RagaAICatalyst(
-    access_key=os.getenv("RAGAAI_CATALYST_ACCESS_KEY"),
-    secret_key=os.getenv("RAGAAI_CATALYST_SECRET_KEY"),
-    base_url=os.getenv("RAGAAI_CATALYST_BASE_URL"),
+    access_key=os.getenv("RAGAAI_CATALYST_PROD_ACCESS_KEY"),
+    secret_key=os.getenv("RAGAAI_CATALYST_PROD_SECRET_KEY"),
+    base_url=os.getenv("RAGAAI_CATALYST_PROD_BASE_URL"),
 )
 
 # Initialize tracer
 tracer = Tracer(
-    project_name="Langgraph_testing",
+    project_name="langgraph_testing_100525",
     dataset_name="customer_support",
     tracer_type="Agentic",
 )
@@ -401,7 +404,7 @@ class State(TypedDict):
 
 
 
-from langchain_anthropic import ChatAnthropic
+# from langchain_anthropic import ChatAnthropic
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.runnables import Runnable, RunnableConfig
@@ -431,14 +434,9 @@ class Assistant:
         return {"messages": result}
 
 
-# Haiku is faster and cheaper, but less accurate
-# llm = ChatAnthropic(model="claude-3-haiku-20240307")
-llm = ChatAnthropic(model="claude-3-5-sonnet-20240620", temperature=1)
-# You could swap LLMs, though you will likely want to update the prompts when
-# doing so!
-# from langchain_openai import ChatOpenAI
+from langchain_openai import ChatOpenAI
 
-# llm = ChatOpenAI(model="gpt-4-turbo-preview")
+llm = ChatOpenAI(model="gpt-4o-mini")
 
 primary_assistant_prompt = ChatPromptTemplate.from_messages(
     [

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/pyproject.toml

@@ -6,9 +6,9 @@ build-backend = "setuptools.build_meta"
 name = "ragaai_catalyst"
 description = "RAGA AI CATALYST"
 readme = "README.md"
-requires-python = ">=3.10,<3.13"
+requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.5.b39"
+version = "2.1.5.b41"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -378,7 +378,7 @@ class AgenticTracing(
             super().add_component(component)
 
         # Handle error case
-        if is_error:
+        if is_error and not self.current_agent_id.get():
             self.stop()
 
     def __enter__(self):

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py

@@ -74,6 +74,10 @@ def extract_model_name(args, kwargs, result):
             return "gemini-1.5-pro"
         if "gemini-pro" in model:
             return "gemini-pro"
+
+    if 'response_metadata' in dir(result):
+        if 'model_name' in result.response_metadata:
+            model = result.response_metadata['model_name']
     
     return model or "default"
 
@@ -116,8 +120,8 @@ def extract_token_usage(result):
         # Run the coroutine in the current event loop
         result = loop.run_until_complete(result)
 
-    # Handle text attribute responses (JSON string or Vertex AI)
-    if hasattr(result, "text"):
+    # Handle text attribute responses (JSON string for Vertex AI)
+    if hasattr(result, "text") and isinstance(result.text, (str, bytes, bytearray)):
         # First try parsing as JSON for OpenAI responses
         try:
             import json
@@ -162,11 +166,26 @@ def extract_token_usage(result):
     # Handle Google GenerativeAI format with usage_metadata
     if hasattr(result, "usage_metadata"):
         metadata = result.usage_metadata
-        return {
-            "prompt_tokens": getattr(metadata, "prompt_token_count", 0),
-            "completion_tokens": getattr(metadata, "candidates_token_count", 0),
-            "total_tokens": getattr(metadata, "total_token_count", 0)
-        }
+        if hasattr(metadata, "prompt_token_count"):
+            return {
+                "prompt_tokens": getattr(metadata, "prompt_token_count", 0),
+                "completion_tokens": getattr(metadata, "candidates_token_count", 0),
+                "total_tokens": getattr(metadata, "total_token_count", 0)
+            }
+        elif hasattr(metadata, "input_tokens"):
+            return {
+                "prompt_tokens": getattr(metadata, "input_tokens", 0),
+                "completion_tokens": getattr(metadata, "output_tokens", 0),
+                "total_tokens": getattr(metadata, "total_tokens", 0)
+            }
+        elif "input_tokens" in metadata:
+            return {
+                "prompt_tokens": metadata["input_tokens"],
+                "completion_tokens": metadata["output_tokens"],
+                "total_tokens": metadata["total_tokens"]
+            }
+
+
     
     # Handle ChatResponse format with raw usuage
     if hasattr(result, "raw") and hasattr(result.raw, "usage"):

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py

@@ -42,7 +42,7 @@ class DynamicTraceExporter(SpanExporter):
         self._project_id = project_id
         self._dataset_name = dataset_name
         self._user_details = user_details
-        self._base_url = base_url,
+        self._base_url = base_url
         self._custom_model_cost = custom_model_cost
 
     
@@ -57,22 +57,37 @@ class DynamicTraceExporter(SpanExporter):
         Returns:
             SpanExportResult: Result of the export operation
         """
-        # Update the exporter's properties
-        self._update_exporter_properties()
-        
-        # Forward the call to the underlying exporter
-        return self._exporter.export(spans)
-    
+        try:
+            # Update the exporter's properties
+            self._update_exporter_properties()
+        except Exception as e:
+            raise Exception(f"Error updating exporter properties: {e}")
+
+        try:
+            # Forward the call to the underlying exporter
+            result = self._exporter.export(spans)
+            return result
+        except Exception as e:
+            raise Exception(f"Error exporting trace: {e}")
+            
+
+
     def shutdown(self):
         """
         Shutdown the exporter by forwarding to the underlying exporter.
         Before shutting down, update the exporter's properties with the current values.
         """
-        # Update the exporter's properties
-        self._update_exporter_properties()
-        
-        # Forward the call to the underlying exporter
-        return self._exporter.shutdown()
+        try:
+            # Update the exporter's properties
+            self._update_exporter_properties()
+        except Exception as e:
+            raise Exception(f"Error updating exporter properties: {e}")
+
+        try:
+            # Forward the call to the underlying exporter
+            return self._exporter.shutdown()
+        except Exception as e:
+            raise Exception(f"Error shutting down exporter: {e}")
     
     def _update_exporter_properties(self):
         """

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -35,6 +35,8 @@ class RAGATraceExporter(SpanExporter):
         for span in spans:
             span_json = json.loads(span.to_json())
             trace_id = span_json.get("context").get("trace_id")
+            if trace_id is None:
+                raise Exception("Trace ID is None")
 
             if trace_id not in self.trace_spans:
                 self.trace_spans[trace_id] = list()
@@ -43,8 +45,14 @@ class RAGATraceExporter(SpanExporter):
 
             if span_json["parent_id"] is None:
                 trace = self.trace_spans[trace_id]
-                self.process_complete_trace(trace, trace_id)
-                del self.trace_spans[trace_id]
+                try:
+                    self.process_complete_trace(trace, trace_id)
+                except Exception as e:
+                    raise Exception(f"Error processing complete trace: {e}")
+                try:
+                    del self.trace_spans[trace_id]
+                except Exception as e:
+                    raise Exception(f"Error deleting trace: {e}")
 
         return SpanExportResult.SUCCESS
 

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41}/ragaai_catalyst/tracers/utils/trace_json_converter.py

@@ -223,7 +223,10 @@ def convert_json_format(input_trace, custom_model_cost):
     }
     final_trace["replays"]={"source":None}
     final_trace["data"]=[{}]
-    final_trace["data"][0]["spans"] = get_spans(input_trace, custom_model_cost)
+    try:
+        final_trace["data"][0]["spans"] = get_spans(input_trace, custom_model_cost)
+    except Exception as e:
+        raise Exception(f"Error in get_spans function: {e}")
     final_trace["network_calls"] =[]
     final_trace["interactions"] = []
 
@@ -231,15 +234,15 @@ def convert_json_format(input_trace, custom_model_cost):
         if itr["type"]=="llm":
             if "tokens" in itr["info"]:
                 if "prompt_tokens" in itr["info"]["tokens"]:
-                    final_trace["metadata"]["tokens"]["prompt_tokens"] += itr["info"]["tokens"]['prompt_tokens']
-                    final_trace["metadata"]["cost"]["input_cost"] += itr["info"]["cost"]['input_cost'] 
+                    final_trace["metadata"]["tokens"]["prompt_tokens"] += itr["info"]["tokens"].get('prompt_tokens', 0.0)
+                    final_trace["metadata"]["cost"]["input_cost"] += itr["info"]["cost"].get('input_cost', 0.0) 
                 if "completion_tokens" in itr["info"]["tokens"]:
-                    final_trace["metadata"]["tokens"]["completion_tokens"] += itr["info"]["tokens"]['completion_tokens']
-                    final_trace["metadata"]["cost"]["output_cost"] += itr["info"]["cost"]['output_cost'] 
+                    final_trace["metadata"]["tokens"]["completion_tokens"] += itr["info"]["tokens"].get('completion_tokens', 0.0)
+                    final_trace["metadata"]["cost"]["output_cost"] += itr["info"]["cost"].get('output_cost', 0.0) 
             if "tokens" in itr["info"]:
                 if "total_tokens" in itr["info"]["tokens"]:
-                    final_trace["metadata"]["tokens"]["total_tokens"] += itr["info"]["tokens"]['total_tokens']
-                    final_trace["metadata"]["cost"]["total_cost"] += itr["info"]["cost"]['total_cost'] 
+                    final_trace["metadata"]["tokens"]["total_tokens"] += itr["info"]["tokens"].get('total_tokens', 0.0)
+                    final_trace["metadata"]["cost"]["total_cost"] += itr["info"]["cost"].get('total_cost', 0.0) 
 
     # get the total tokens, cost
     final_trace["metadata"]["total_cost"] = final_trace["metadata"]["cost"]["total_cost"] 

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b41/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b39
+Version: 2.1.5b41
 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>, Tushar Kumar <tushar.kumar@raga.ai>
-Requires-Python: <3.13,>=3.10
+Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2