Skip to content

Commit f58a60b

Browse files
committed
Add assertions on the gen_ai.system within new langchain tests
1 parent 5cc615e commit f58a60b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/integrations/langchain/test_langchain.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)