Merge branch 'webb/langchain/add-basic-test' into webb/langchain/tool-test

Author: alexander-alderman-webb

tests/integrations/langchain/test_langchain.py

@@ -136,7 +136,7 @@ def test_langchain_create_agent(
     )
     events = capture_events()
 
-    model_repsonse = get_model_response(
+    model_response = get_model_response(
         nonstreaming_responses_model_response,
         serialize_pydantic=True,
         request_headers={
@@ -160,7 +160,7 @@ def test_langchain_create_agent(
     with patch.object(
         llm.client._client._client,
         "send",
-        return_value=model_repsonse,
+        return_value=model_response,
     ) as _:
         with start_transaction():
             agent.invoke(
@@ -179,12 +179,9 @@ def test_langchain_create_agent(
     assert len(chat_spans) == 1
     assert chat_spans[0]["origin"] == "auto.ai.langchain"
 
-    # Token usage is only available in newer versions of langchain (v0.2+)
-    # where usage_metadata is supported on AIMessageChunk
-    if "gen_ai.usage.input_tokens" in chat_spans[0]["data"]:
-        assert chat_spans[0]["data"]["gen_ai.usage.input_tokens"] == 10
-        assert chat_spans[0]["data"]["gen_ai.usage.output_tokens"] == 20
-        assert chat_spans[0]["data"]["gen_ai.usage.total_tokens"] == 30
+    assert chat_spans[0]["data"]["gen_ai.usage.input_tokens"] == 10
+    assert chat_spans[0]["data"]["gen_ai.usage.output_tokens"] == 20
+    assert chat_spans[0]["data"]["gen_ai.usage.total_tokens"] == 30
 
     if send_default_pii and include_prompts:
         assert (