ragaai-catalyst 2.1.2b3__py3-none-any.whl → 2.1.2b4__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py

@@ -185,9 +185,25 @@ class LLMTracerMixin:
         @functools.wraps(original_init)
         def patched_init(client_self, *args, **kwargs):
             original_init(client_self, *args, **kwargs)
-            self.wrap_method(client_self.chat.completions, "create")
-            if hasattr(client_self.chat.completions, "acreate"):
-                self.wrap_method(client_self.chat.completions, "acreate")
+            # Check if this is AsyncOpenAI or OpenAI
+            is_async = 'AsyncOpenAI' in client_class.__name__
+            
+            if is_async:
+                # Patch async methods for AsyncOpenAI
+                if hasattr(client_self.chat.completions, "create"):
+                    original_create = client_self.chat.completions.create
+                    @functools.wraps(original_create)
+                    async def wrapped_create(*args, **kwargs):
+                        return await self.trace_llm_call(original_create, *args, **kwargs)
+                    client_self.chat.completions.create = wrapped_create
+            else:
+                # Patch sync methods for OpenAI
+                if hasattr(client_self.chat.completions, "create"):
+                    original_create = client_self.chat.completions.create
+                    @functools.wraps(original_create)
+                    def wrapped_create(*args, **kwargs):
+                        return self.trace_llm_call_sync(original_create, *args, **kwargs)
+                    client_self.chat.completions.create = wrapped_create
 
         setattr(client_class, "__init__", patched_init)
 
@@ -495,12 +511,12 @@ class LLMTracerMixin:
             raise
 
     def trace_llm(self, name: str = None):
-        self.current_llm_call_name.set(name)
         def decorator(func):
             @self.file_tracker.trace_decorator
             @functools.wraps(func)
             async def async_wrapper(*args, **kwargs):
                 self.gt = kwargs.get('gt', None) if kwargs else None
+                self.current_llm_call_name.set(name)
                 if not self.is_active:
                     return await func(*args, **kwargs)
                 
@@ -550,6 +566,7 @@ class LLMTracerMixin:
             @functools.wraps(func)
             def sync_wrapper(*args, **kwargs):
                 self.gt = kwargs.get('gt', None) if kwargs else None
+                self.current_llm_call_name.set(name)
                 if not self.is_active:
                     return func(*args, **kwargs)
                 

{ragaai_catalyst-2.1.2b3.dist-info → ragaai_catalyst-2.1.2b4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.2b3
+Version: 2.1.2b4
 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.13,>=3.9

{ragaai_catalyst-2.1.2b3.dist-info → ragaai_catalyst-2.1.2b4.dist-info}/RECORD

@@ -28,7 +28,7 @@ ragaai_catalyst/tracers/agentic_tracing/tests/unique_decorator_test.py,sha256=Xk
 ragaai_catalyst/tracers/agentic_tracing/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=FOKJUMd48s8NFoTiZINeWBPA4IjMo5o1r9nuDEzIerA,20945
 ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=liLdiHdqc6BHH4YyvrdmX_fprXoeJchOqBwmzMJPrCc,14075
-ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=j3-aie4aXT7dVciOdsCU6aiI4nEZ6y4yYJAbrEXgCtM,24484
+ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=EWcse5vJdOO5a3uSuEL6TKRsnvKIW_H9qbUVgRTxq1M,25449
 ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=tdwkpWx_eSyXtbpWTo9RG_cF_6q1F8oB0XyF4OiRKh8,10363
 ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=6FTA15xMnum9omM_0Jd9cMIuWdKu1gR5Tc8fOXAkP8E,10068
 ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=Rln82_4BEL_4ULOIhWwu2-qGQ8J1yEaf9qKsEBuj76o,8427
@@ -54,7 +54,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-2.1.2b3.dist-info/METADATA,sha256=Rp-1XiBpknVy7_4rWG9d7KKTV8ZBSoh7w78AT9uwYu4,1804
-ragaai_catalyst-2.1.2b3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.2b3.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.2b3.dist-info/RECORD,,
+ragaai_catalyst-2.1.2b4.dist-info/METADATA,sha256=yrtCVe68YRRwrejlCWGHpRGDo2e9Vh6dAnHJHGJvS4M,1804
+ragaai_catalyst-2.1.2b4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.2b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.2b4.dist-info/RECORD,,