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.
- ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py +21 -4
- {ragaai_catalyst-2.1.2b3.dist-info → ragaai_catalyst-2.1.2b4.dist-info}/METADATA +1 -1
- {ragaai_catalyst-2.1.2b3.dist-info → ragaai_catalyst-2.1.2b4.dist-info}/RECORD +5 -5
- {ragaai_catalyst-2.1.2b3.dist-info → ragaai_catalyst-2.1.2b4.dist-info}/WHEEL +0 -0
- {ragaai_catalyst-2.1.2b3.dist-info → ragaai_catalyst-2.1.2b4.dist-info}/top_level.txt +0 -0
@@ -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
|
-
|
189
|
-
|
190
|
-
|
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.
|
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=
|
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.
|
58
|
-
ragaai_catalyst-2.1.
|
59
|
-
ragaai_catalyst-2.1.
|
60
|
-
ragaai_catalyst-2.1.
|
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,,
|
File without changes
|
File without changes
|