Skip to content

Commit 23ee128

Browse files
gustavocidornelaswhoseoyster
authored andcommitted
fix(openai tracer): Azure OpenAI chat completion step duplicated
1 parent af49b20 commit 23ee128

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

src/openlayer/lib/integrations/openai_tracer.py

+7-6
Original file line numberDiff line numberDiff line change
@@ -137,9 +137,9 @@ def stream_chunks(
137137
if delta.function_call.name:
138138
collected_function_call["name"] += delta.function_call.name
139139
if delta.function_call.arguments:
140-
collected_function_call["arguments"] += (
141-
delta.function_call.arguments
142-
)
140+
collected_function_call[
141+
"arguments"
142+
] += delta.function_call.arguments
143143
elif delta.tool_calls:
144144
if delta.tool_calls[0].function.name:
145145
collected_function_call["name"] += delta.tool_calls[0].function.name
@@ -257,9 +257,10 @@ def add_to_trace(is_azure_openai: bool = False, **kwargs) -> None:
257257
tracer.add_chat_completion_step_to_trace(
258258
**kwargs, name="Azure OpenAI Chat Completion", provider="Azure"
259259
)
260-
tracer.add_chat_completion_step_to_trace(
261-
**kwargs, name="OpenAI Chat Completion", provider="OpenAI"
262-
)
260+
else:
261+
tracer.add_chat_completion_step_to_trace(
262+
**kwargs, name="OpenAI Chat Completion", provider="OpenAI"
263+
)
263264

264265

265266
def handle_non_streaming_create(

0 commit comments

Comments
 (0)