@@ -159,6 +159,7 @@ def test_langchain_text_completion(
159159
160160 llm_span = llm_spans [0 ]
161161 assert llm_span ["description" ] == "generate_text gpt-3.5-turbo"
162+ assert llm_span ["data" ]["gen_ai.system" ] == "openai"
162163 assert llm_span ["data" ]["gen_ai.request.model" ] == "gpt-3.5-turbo"
163164 assert llm_span ["data" ]["gen_ai.response.text" ] == "The capital of France is Paris."
164165 assert llm_span ["data" ]["gen_ai.usage.total_tokens" ] == 25
@@ -254,6 +255,7 @@ def test_langchain_create_agent(
254255 assert len (chat_spans ) == 1
255256 assert chat_spans [0 ]["origin" ] == "auto.ai.langchain"
256257
258+ assert chat_spans [0 ]["data" ]["gen_ai.system" ] == "openai-chat"
257259 assert chat_spans [0 ]["data" ]["gen_ai.usage.input_tokens" ] == 10
258260 assert chat_spans [0 ]["data" ]["gen_ai.usage.output_tokens" ] == 20
259261 assert chat_spans [0 ]["data" ]["gen_ai.usage.total_tokens" ] == 30
@@ -413,10 +415,12 @@ def test_tool_execution_span(
413415 assert chat_spans [0 ]["data" ]["gen_ai.usage.input_tokens" ] == 142
414416 assert chat_spans [0 ]["data" ]["gen_ai.usage.output_tokens" ] == 50
415417 assert chat_spans [0 ]["data" ]["gen_ai.usage.total_tokens" ] == 192
418+ assert chat_spans [0 ]["data" ]["gen_ai.system" ] == "openai-chat"
416419
417420 assert chat_spans [1 ]["data" ]["gen_ai.usage.input_tokens" ] == 89
418421 assert chat_spans [1 ]["data" ]["gen_ai.usage.output_tokens" ] == 28
419422 assert chat_spans [1 ]["data" ]["gen_ai.usage.total_tokens" ] == 117
423+ assert chat_spans [1 ]["data" ]["gen_ai.system" ] == "openai-chat"
420424
421425 if send_default_pii and include_prompts :
422426 assert "word" in tool_exec_span ["data" ][SPANDATA .GEN_AI_TOOL_INPUT ]
0 commit comments