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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -185,9 +185,25 @@ class LLMTracerMixin:
185
185
  @functools.wraps(original_init)
186
186
  def patched_init(client_self, *args, **kwargs):
187
187
  original_init(client_self, *args, **kwargs)
188
- self.wrap_method(client_self.chat.completions, "create")
189
- if hasattr(client_self.chat.completions, "acreate"):
190
- self.wrap_method(client_self.chat.completions, "acreate")
188
+ # Check if this is AsyncOpenAI or OpenAI
189
+ is_async = 'AsyncOpenAI' in client_class.__name__
190
+
191
+ if is_async:
192
+ # Patch async methods for AsyncOpenAI
193
+ if hasattr(client_self.chat.completions, "create"):
194
+ original_create = client_self.chat.completions.create
195
+ @functools.wraps(original_create)
196
+ async def wrapped_create(*args, **kwargs):
197
+ return await self.trace_llm_call(original_create, *args, **kwargs)
198
+ client_self.chat.completions.create = wrapped_create
199
+ else:
200
+ # Patch sync methods for OpenAI
201
+ if hasattr(client_self.chat.completions, "create"):
202
+ original_create = client_self.chat.completions.create
203
+ @functools.wraps(original_create)
204
+ def wrapped_create(*args, **kwargs):
205
+ return self.trace_llm_call_sync(original_create, *args, **kwargs)
206
+ client_self.chat.completions.create = wrapped_create
191
207
 
192
208
  setattr(client_class, "__init__", patched_init)
193
209
 
@@ -495,12 +511,12 @@ class LLMTracerMixin:
495
511
  raise
496
512
 
497
513
  def trace_llm(self, name: str = None):
498
- self.current_llm_call_name.set(name)
499
514
  def decorator(func):
500
515
  @self.file_tracker.trace_decorator
501
516
  @functools.wraps(func)
502
517
  async def async_wrapper(*args, **kwargs):
503
518
  self.gt = kwargs.get('gt', None) if kwargs else None
519
+ self.current_llm_call_name.set(name)
504
520
  if not self.is_active:
505
521
  return await func(*args, **kwargs)
506
522
 
@@ -550,6 +566,7 @@ class LLMTracerMixin:
550
566
  @functools.wraps(func)
551
567
  def sync_wrapper(*args, **kwargs):
552
568
  self.gt = kwargs.get('gt', None) if kwargs else None
569
+ self.current_llm_call_name.set(name)
553
570
  if not self.is_active:
554
571
  return func(*args, **kwargs)
555
572
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: ragaai_catalyst
3
- Version: 2.1.2b3
3
+ Version: 2.1.2b4
4
4
  Summary: RAGA AI CATALYST
5
5
  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>
6
6
  Requires-Python: <3.13,>=3.9
@@ -28,7 +28,7 @@ ragaai_catalyst/tracers/agentic_tracing/tests/unique_decorator_test.py,sha256=Xk
28
28
  ragaai_catalyst/tracers/agentic_tracing/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
29
  ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=FOKJUMd48s8NFoTiZINeWBPA4IjMo5o1r9nuDEzIerA,20945
30
30
  ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=liLdiHdqc6BHH4YyvrdmX_fprXoeJchOqBwmzMJPrCc,14075
31
- ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=j3-aie4aXT7dVciOdsCU6aiI4nEZ6y4yYJAbrEXgCtM,24484
31
+ ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=EWcse5vJdOO5a3uSuEL6TKRsnvKIW_H9qbUVgRTxq1M,25449
32
32
  ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=tdwkpWx_eSyXtbpWTo9RG_cF_6q1F8oB0XyF4OiRKh8,10363
33
33
  ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=6FTA15xMnum9omM_0Jd9cMIuWdKu1gR5Tc8fOXAkP8E,10068
34
34
  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
54
54
  ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
55
55
  ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
56
56
  ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
57
- ragaai_catalyst-2.1.2b3.dist-info/METADATA,sha256=Rp-1XiBpknVy7_4rWG9d7KKTV8ZBSoh7w78AT9uwYu4,1804
58
- ragaai_catalyst-2.1.2b3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
59
- ragaai_catalyst-2.1.2b3.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
60
- ragaai_catalyst-2.1.2b3.dist-info/RECORD,,
57
+ ragaai_catalyst-2.1.2b4.dist-info/METADATA,sha256=yrtCVe68YRRwrejlCWGHpRGDo2e9Vh6dAnHJHGJvS4M,1804
58
+ ragaai_catalyst-2.1.2b4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
59
+ ragaai_catalyst-2.1.2b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
60
+ ragaai_catalyst-2.1.2b4.dist-info/RECORD,,