langtrace-python-sdk 2.0.4__py3-none-any.whl → 2.0.6__py3-none-any.whl

{langtrace_python_sdk-2.0.4.dist-info → langtrace_python_sdk-2.0.6.dist-info}/RECORD

@@ -1,43 +1,43 @@
 examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/anthropic_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/anthropic_example/completion.py,sha256=g7xiDDlnA-bc_tbfOsDVeCOMja-y_9v9ByYKhK83Rt4,717
+examples/anthropic_example/completion.py,sha256=PVxAth2vqbX_QN2DGVOt2mTGuUiGk4O5w7RFhHzsxA8,714
 examples/chroma_example/__init__.py,sha256=Tq6pae7fHA7dwuDslabB5MTNedL21gu2RaZicpxSyLU,25
-examples/chroma_example/basic.py,sha256=RT4yi8mhN_bgTbeGZnHEscLT9z_VlFnHekIM9psefLk,870
+examples/chroma_example/basic.py,sha256=qngW9EQtIeuBFjyWdPPiJAE1xnVimKqkglJQz9cqueM,867
 examples/cohere_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/cohere_example/chat.py,sha256=iPxOp8BGyBreHw0_pZRInUPwubALVvAJbWecdvy3sNU,868
-examples/cohere_example/chat_stream.py,sha256=SEcE8dTemOP8XZUxd08gn8nnBNlZfWa2t4tqnmmC7_Y,609
-examples/cohere_example/embed.py,sha256=PzhEBNUdBlHKXBT95c3wp5GFtNvPu58ywHlcPE3MkVE,525
-examples/cohere_example/rerank.py,sha256=mFCX4nNfKDmxqyCPqWCWqKjRS0XXH4M9txT-tzcKliw,1147
-examples/cohere_example/tools.py,sha256=kBN3cqMeBy-jOlSnFszx0bud6niEy1SsBehLMimBWVw,1434
-examples/fastapi_example/basic_route.py,sha256=O0gaAdYgdbZ1LElV5JuJyq-FZMHcfJCysqQPpKa8fNI,1128
-examples/hiveagent_example/basic.py,sha256=MuzPMHtrhkxv2T7Ke8Rz3TsSum_XZ0IyPc2ufOt0U1Y,444
+examples/cohere_example/chat.py,sha256=0iNVE3sxX6fZBSRozzx9a8M7HI4QtP1BIjNguBO32rw,915
+examples/cohere_example/chat_stream.py,sha256=9hT5whG_W002jxHSXgpQR6imaoQJ10ydte3cFuXc8fE,638
+examples/cohere_example/embed.py,sha256=9yNyl2_aJun-0n1v4B_H6PlCeO7LQDIuMagRzwC3OWs,523
+examples/cohere_example/rerank.py,sha256=Zy1H7TwXHOGulYAMkfOe-hB0YmJIfwfPurYxj__l_98,1145
+examples/cohere_example/tools.py,sha256=I_uCDHQiLRD4uCmeXRUTYNtTgO_efy809Pik1j-Usfw,1683
+examples/fastapi_example/basic_route.py,sha256=VXb9r8usZRqQOwS4bzonsP0jFYhvfZecbK8zYcOxAl4,1125
+examples/hiveagent_example/basic.py,sha256=Sd7I5w8w5Xx7ODaydTY30yiq9HwJDMKHQywrZjgehP0,441
 examples/langchain_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/langchain_example/basic.py,sha256=mSVcE-Kx7jNS4U_zrm5WZI98xyw6WTrhd2GMVGgn9Cg,2653
-examples/langchain_example/groq_example.py,sha256=GaQTufs4UPoWAimNbeH1wFGkM6NcSEfbzHy1jE2EL8M,773
-examples/langchain_example/langgraph_example.py,sha256=h9tuPrLJ857U7Iz0RshW1LWbPjWYco-WvEP7t-xnJZY,1998
+examples/langchain_example/groq_example.py,sha256=9dBbOWZCVJ29PpB1NbJhc9MThmtwrlOpdNIxPl7d8f8,787
+examples/langchain_example/langgraph_example.py,sha256=7C2a4Sg0PKbbab03CVkStO3MzT7C-O1UtdmObvBXurM,2005
 examples/langchain_example/tool.py,sha256=8T8_IDbgA58XbsfyH5_xhA8ZKQfyfyFxF8wor-PsRjA,2556
 examples/llamaindex_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/llamaindex_example/agent.py,sha256=_iIXy9lfDz6ySf6aTeeRqejlfGnXZ7msxLBjGiJth3E,2777
-examples/llamaindex_example/basic.py,sha256=gvns3oDUy0c4I5ewnj9-B36_1La8y6qD3VQaq6v3syM,654
+examples/llamaindex_example/agent.py,sha256=41uxHpMKqIB4irD8YHu6Y0QI3pDfz79huUHroQwhfDA,2774
+examples/llamaindex_example/basic.py,sha256=kcl2RykIzq1Lv9KePzp_TCcdp_PFXxt3vsspb5p-5HI,651
 examples/llamaindex_example/data/abramov.txt,sha256=Ou-GyWZm5AjHLgxviBoRE9ikNv5MScsF0cd--0vVVhI,32667
 examples/openai_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/openai_example/async_tool_calling_nonstreaming.py,sha256=foomPKwpju0PMazdE3xNrdZWguUYMnNXibqq01-uBvc,3806
-examples/openai_example/async_tool_calling_streaming.py,sha256=TayOsb0jcYYlFpnWWUYxwCVKuJhTfGfE8LkjYJNGSz4,6900
-examples/openai_example/chat_completion.py,sha256=ib5lj89lnDBK2iErsQmBTlsWMbRUxyoqNQDIKHGrHVI,1235
-examples/openai_example/embeddings_create.py,sha256=fqmTXdzqFabUbtu9dzoL-TR0fEURnv8zI_oLQoX_l6o,504
-examples/openai_example/function_calling.py,sha256=wyGMlvTPnBe4MFGcnMt-B0_cmiVmd5pEBbr6ATexxaA,2187
-examples/openai_example/images_generate.py,sha256=xV9XAJnax4TmCXsyDnzSL9zhh2VtojwpfWf0k7nGCOQ,472
-examples/openai_example/tool_calling.py,sha256=M2zq2zW_qXQr1GlAUePnglICZTzC5j5QA8hTzyHnOi4,2319
-examples/openai_example/tool_calling_nonstreaming.py,sha256=RrJaLbbs82bkCZ8tIu6u9re95gDDgUb6nT3TYbDBTn0,3778
-examples/openai_example/tool_calling_streaming.py,sha256=WnWWlgDqKuqN2DtWbpJs_JvmmQehBZp0Ke3ZXvCJdQw,6860
-examples/perplexity_example/basic.py,sha256=oTLwEYlvpD4wEnqEUrUSlQ0SeQ0u50Jeab4ggkikQg0,671
+examples/openai_example/async_tool_calling_nonstreaming.py,sha256=7mP6Anonj7B8owPBfbhxyux4mdgp3XCuym1xAx6SLls,3874
+examples/openai_example/async_tool_calling_streaming.py,sha256=PQMHSWAN1DfgjK88b8LVSredmov0uH9Fgn1FHoCVv1M,7054
+examples/openai_example/chat_completion.py,sha256=f2rMHJU1Dph_fAOv9qCksvtldl1s9UitEeLsvyoSpGA,1253
+examples/openai_example/embeddings_create.py,sha256=d05rArTWxsmCxOdTRaGw3n9Yt1HXxOHHCtciCHIwBGc,502
+examples/openai_example/function_calling.py,sha256=rrnrFw2HyV_gv4TdwUr299baP4tIupz6EQWWmLDtJkU,2265
+examples/openai_example/images_generate.py,sha256=SZNY8Visk7JUpx5QhNxTNINHmPAGdCUayF-Q7_iCr50,470
+examples/openai_example/tool_calling.py,sha256=_IV7KoSI_37u1TTZWdVa58BYjkDfhSurvM86xwaNNhY,2316
+examples/openai_example/tool_calling_nonstreaming.py,sha256=Yc848IooZRXNynHL6z0kOgJ4qbmL_NOufcb2VmWRukI,3847
+examples/openai_example/tool_calling_streaming.py,sha256=mV1RbyAoVhumGRPpqPWQ6PMhnJyeifrlELd2-K1qJ_w,7015
+examples/perplexity_example/basic.py,sha256=bp7n27gaugJkaFVyt8pjaEfi66lYcqP6eFFjPewUShY,668
 examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/pinecone_example/basic.py,sha256=OkYjN3J5kxw-kloOV3Q-iyI6opkbarWsMom-_AMP2ZA,893
+examples/pinecone_example/basic.py,sha256=9IWMoMLozMTpxjY8BFBg2EpUNY5dDjU1xocEG9XsXP0,966
 examples/qdrant_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/qdrant_example/basic.py,sha256=2JDMsBllnRvVDuuZ91FkAkpXgnavUP4mqXa51CtfHP8,1421
+examples/qdrant_example/basic.py,sha256=DCMjHSuBZKkhEjCkwy5d5La9WMyW0lCWqtcZWiFCEm4,1425
 langtrace_python_sdk/__init__.py,sha256=q2TAsxGIv4wFIQA7G-HGxlnzO89bhPU5TgI42sPHLj0,740
-langtrace_python_sdk/langtrace.py,sha256=HuGbzXM_oqIzpfFDyV2nxuEe4C3SCxDlS1qFzswXzY4,4762
-langtrace_python_sdk/version.py,sha256=ZOg41aCgKEuObK5WhegikXektFWPJaeZ-TMDh0Ke95M,22
+langtrace_python_sdk/langtrace.py,sha256=z6CL_e2aQUXJlTUzel0IL2zBVAa5lxRR01b88GfaBuY,6576
+langtrace_python_sdk/version.py,sha256=4Yxft_MNOtwQG58mZpa8eoM0XBHa8VNkTPH9_i0gDaM,22
 langtrace_python_sdk/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=5MNjnAOg-4am78J3gVMH6FSwq5N8TOj72ugkhsw4vi0,46
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -45,52 +45,53 @@ langtrace_python_sdk/constants/instrumentation/anthropic.py,sha256=YX3llt3zwDY6X
 langtrace_python_sdk/constants/instrumentation/chroma.py,sha256=hiPGYdHS0Yj4Kh3eaYBbuCAl_swqIygu80yFqkOgdak,955
 langtrace_python_sdk/constants/instrumentation/cohere.py,sha256=tf9sDfb5K3qOAHChEE5o8eYWPZ1io58VsOjZDCZPxfw,577
 langtrace_python_sdk/constants/instrumentation/common.py,sha256=nrWGQf6m-GV_v8-RGbaFrrEeUWZ6pNcYzUWCy-PoQpc,726
-langtrace_python_sdk/constants/instrumentation/groq.py,sha256=gi-l7hIczdkpjCV8-9VqxY9tQLHEK-As3WESlP8gkus,136
+langtrace_python_sdk/constants/instrumentation/groq.py,sha256=VFXmIl4aqGY_fS0PAmjPj_Qm7Tibxbx7Ur_e7rQpqXc,134
 langtrace_python_sdk/constants/instrumentation/openai.py,sha256=9VF6ic9Ed3bpSvdp6iNmrpx2Ppo6DPav3hoUcqSQSv0,1048
 langtrace_python_sdk/constants/instrumentation/pinecone.py,sha256=Xaqqw-xBO0JJLGk75hiCUQGztNm0HiVaLQvjtYK7VJM,472
 langtrace_python_sdk/constants/instrumentation/qdrant.py,sha256=yL7BopNQTXW7L7Z-gVM2PdusKD7r9qqcATvczFd7NtQ,1999
 langtrace_python_sdk/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=3KKeWSLECAxO-FMTndwF5Evv39t9-GG7p67DfYJpNEY,4194
+langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=s-4zc7lba1va_JUReFdLgyBKlvwV1KmBt_OzVmWReQg,3610
 langtrace_python_sdk/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=Tum7OhRRim2cTxlMdYk3iFjWTtQdx9HBd8TtbZ02l64,1690
-langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=TU9Z8f7Yniu7NCBoqKHF__V1dUF6fJOKVvvw4ZHGBS8,8249
+langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=-srgE8qumAn0ulQYZxMa8ch-9IBH0XgBW_rfEnGk6LI,1684
+langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=koAo6bA6nnRDk01fcrArqnyAmgzjxV20ofITMQyKVws,8427
 langtrace_python_sdk/instrumentation/chroma/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/chroma/instrumentation.py,sha256=nT6PS6bsrIOO9kLV5GuUeRjMe6THHHAZGvqWBP1dYog,1807
-langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=gGBX-zLQcOWUWt596PudOJftZcHge0mA_X1ltW3C800,2802
+langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=qytjkZxntUe7XyAvVoiEvFgUIXNU52o9BnYLAnQ9M20,2809
 langtrace_python_sdk/instrumentation/cohere/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/cohere/instrumentation.py,sha256=XAs95fPBh2TTd6E1qUbi-FUg7PClW3Y6P1eOq2f38VA,2228
+langtrace_python_sdk/instrumentation/cohere/instrumentation.py,sha256=YQFHZIBd7SSPD4b6Va-ZR0thf_AuBCqj5yzHLHJVWnM,2121
 langtrace_python_sdk/instrumentation/cohere/patch.py,sha256=hdSWArP0cdfGs0Q5KhkaowkwrYQIzxvF8NpY4sg8p2Q,26563
 langtrace_python_sdk/instrumentation/groq/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/groq/instrumentation.py,sha256=LHzJkICjflnvAhRzEM7s-Pd9tRKlI3qLOYg0_vYDos0,1826
-langtrace_python_sdk/instrumentation/groq/patch.py,sha256=uuEAPZQXGmvuMAb6lIV-1elz9kNe7L3wDwIASUjrQAQ,25572
+langtrace_python_sdk/instrumentation/groq/instrumentation.py,sha256=Ttf07XVKhdYY1_fqJc7QWiSdmgEhEVyQB_3Az2_wqYo,1832
+langtrace_python_sdk/instrumentation/groq/patch.py,sha256=Fo5QbNwScFwhvmXZQezbjF0AdajsHQ8a0f8nKsVFjF8,26068
 langtrace_python_sdk/instrumentation/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/langchain/instrumentation.py,sha256=_Z4AeNb2hBPSCvMRxE-mUfmkUO_wP_tGGtu-jppWPiI,3462
-langtrace_python_sdk/instrumentation/langchain/patch.py,sha256=NdDeSTAxPvUdWkJS1ntwNXmS5o7vtNhWqgYOxm9IsjA,3789
+langtrace_python_sdk/instrumentation/langchain/patch.py,sha256=2BT_-DjfSPTmyHDqnpj-gVrMFxRSZugVC-YRPXnuupU,3796
 langtrace_python_sdk/instrumentation/langchain_community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/langchain_community/instrumentation.py,sha256=6mNHzZwfLOAswYykRbxDBUeqVEx3ZIyf6iKxRxQe3jM,5198
-langtrace_python_sdk/instrumentation/langchain_community/patch.py,sha256=-uuFDk3OdrgBv2oVyYWpyCgzIhfJRpt04HgLHcsu3Vk,3745
+langtrace_python_sdk/instrumentation/langchain_community/instrumentation.py,sha256=-9E2-8-XR5PQ5ON-_ib53p5Tp0uD4MT4Ioh4MlWlF2k,5192
+langtrace_python_sdk/instrumentation/langchain_community/patch.py,sha256=uyRUp02M9mYjLsr36ZBCvJC7s40AGry3R_EcjvXaoWk,3751
 langtrace_python_sdk/instrumentation/langchain_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/langchain_core/instrumentation.py,sha256=pYEDl5XD2jlIb7BqNX89eZRqMn05PXy0UhpuyQPNTOM,5931
-langtrace_python_sdk/instrumentation/langchain_core/patch.py,sha256=8WspiRAIV_pgC8fLfdmjQqgjvoHXvudx-FL9qC6L9mw,8451
+langtrace_python_sdk/instrumentation/langchain_core/instrumentation.py,sha256=KNXHtlnq80akLACF-vbPCo9oVXhnVjYIM1Lo7kNRwEg,5937
+langtrace_python_sdk/instrumentation/langchain_core/patch.py,sha256=Kqpv77ZKrZ4QMit36JtN59ubtWE8pX6t8gg6OOmIii0,8458
 langtrace_python_sdk/instrumentation/langgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/langgraph/instrumentation.py,sha256=C26-sSFQSohFgcooeisFz3jj1rEhrwemRyLwFa4qoFw,2374
-langtrace_python_sdk/instrumentation/langgraph/patch.py,sha256=HDz4twVGKDWtR0vggikhKGPglAOOXOV33ktyH_Yghl8,4043
+langtrace_python_sdk/instrumentation/langgraph/instrumentation.py,sha256=SUZZhWSIbcfsF1S5NtEqW8QzkRM_pKAuXB7pwk5tsOU,2526
+langtrace_python_sdk/instrumentation/langgraph/patch.py,sha256=fr90kuPokwxP2BGP2tU793ReDOp2FtehKMcsxS9tS_0,4589
 langtrace_python_sdk/instrumentation/llamaindex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=8iAg-Oxwf2W4S60qRfO5mvzORYxublgq7FdGWqUB4q8,2965
-langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=oytT6EujmCd1Zf5ccde5juiucnDpYiIHjz-jan284Zk,4183
+langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=q99qmEBNH54Oi_sHCIGBQFpIE4zaNKVjIWARlX8fJ-M,4189
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=kXu8zmppMepYvXRoVG9ERx__0xfSQ8uyPPWTihmrlAk,2712
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=UYEA23SO3XDJbhIXAkZ5xAiBuItwSjebGL45RsKNJdo,36645
+langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=G2HSZfr6DuP2n-8v0h91M60m0DJWFBcru4-1FQJl-5A,2726
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=kQxc_E1IlOgHZnlc0HbpGjM5rZCDR3YpNFChp2X9GbM,37263
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=52GbkfRK-sxXBXhHmHRRO2BGNd6zZVHDp8njLS1CQX0,2314
-langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=WweBK2-9PbnmwNVN4EnxbriM9CDL079RonN5ZbpBRIE,2698
+langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=ebz57WgmlTnY-FDwQgJ-_eqsknEmDWfh-AiNow4G94s,2705
 langtrace_python_sdk/instrumentation/qdrant/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/qdrant/instrumentation.py,sha256=vl2eKSP55aqDo1JiRlvOUBrr6kddvG9Z5dCYew2OG08,1816
-langtrace_python_sdk/instrumentation/qdrant/patch.py,sha256=acT3Jqibsab_usHKRUc61CmXWEeH-mKyTxZiK2oUO3k,2800
+langtrace_python_sdk/instrumentation/qdrant/patch.py,sha256=skVziJbvoR7XYTI6KmsiQlclTmQEeLeJ5UrJLvi7fjw,2807
+langtrace_python_sdk/types/__init__.py,sha256=QYK-kRkyfXDdzjjKmXVxi0vZ6Wpbxdf0CoZbTXO0JT8,787
 langtrace_python_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/utils/llm.py,sha256=KA5ZL397_ktSpAnIOpxYTaSjaaAoxdHYfiAzbbfFfNA,2022
-langtrace_python_sdk/utils/with_root_span.py,sha256=FOiCV59IAhO861ftbwSCIbU9d4a2EBBjxsqc5HPlZf8,2452
+langtrace_python_sdk/utils/llm.py,sha256=Cl710-0rfr15g1xNkOpLpgCn84pNlfMflcQ4S5jSRu0,2010
+langtrace_python_sdk/utils/with_root_span.py,sha256=MXF-FmG47P0GtJQNffKM_3XGrJHVAC4hMB6GmwtnA4M,2454
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/conftest.py,sha256=0Jo6iCZTXbdvyJVhG9UpYGkLabL75378oauCzmt-Sa8,603
 tests/utils.py,sha256=hP8sTH-M8WO6zlLkSFHPf6483ZcKEcnxul6JIIb1pLM,1396
@@ -99,7 +100,7 @@ tests/anthropic/test_anthropic.py,sha256=p4WhcECJ_zctirtkAX6WejrlNfZzo90fSSxqQki
 tests/anthropic/cassettes/test_anthropic.yaml,sha256=z7YAqA_BBgI-hw7uyVMdLoIZdBYhKwl9cuTzUu9nAZs,2332
 tests/anthropic/cassettes/test_anthropic_streaming.yaml,sha256=D0w4-6dfsgrhbNLJEj8gZBV0yXfrAfA9u90Yu8Ac-TY,11675
 tests/anthropic/cassettes/test_async_anthropic_streaming.yaml,sha256=hQZPY2vwBaW3BWllLd0lcGQ73DjA8C3Ips1Hx9pA-ao,8373
-tests/chroma/test_chroma.py,sha256=5zeInDcP5VplrM9ex2iuVKcpVKMDfEf_ZDK9D6Tc700,2392
+tests/chroma/test_chroma.py,sha256=bVWC_OwmarTgVzKxZy5lrXpJzu9v9Ko6uQdktxOVh74,2430
 tests/cohere/conftest.py,sha256=jBbyg-tut-ZJN5_5D64sGmaPIhT_nQQQAiW43kl5Rz4,621
 tests/cohere/test_cohere_chat.py,sha256=MioEIXB5e-4t99Vt-KpLlxd4diPbp3S6eNfnPWjEDXM,4564
 tests/cohere/test_cohere_embed.py,sha256=bFKDzUnypMyLefAOHR-6dEgvt0yys2Yw7-F0zc0uQyc,1250
@@ -108,9 +109,9 @@ tests/cohere/cassettes/test_cohere_chat.yaml,sha256=_Az6Kfhp1XmoK6zwTxqbnRYTmsm0
 tests/cohere/cassettes/test_cohere_chat_streaming.yaml,sha256=TdWe64Pyfa5Ho9lqTMIW41YntUmUwmYQc55qWntbvas,10610
 tests/cohere/cassettes/test_cohere_embed.yaml,sha256=YAxJ3otY9N6jJQOKi50nQA4xY9-dknm1oXEP8KiCH5U,27312
 tests/cohere/cassettes/test_cohere_rerank.yaml,sha256=m5EAKAM5JEcGMyUlmjaxU-cNLiXKcenSFQyU-UaUI0U,2392
-tests/langchain/test_langchain.py,sha256=GGGRcxz0isNmegeum37XFrlJqI6jB6_iUvv8AJ5iG24,2481
-tests/langchain/test_langchain_community.py,sha256=m9lBmMZIeUouKSq1JfdBupV0-0ef39GD6BKsA0Cf_08,2515
-tests/langchain/test_langchain_core.py,sha256=hCuKkIUvDQOUBM2oEgMG3Iq_KNTwC2sH7_Y_IR5FIno,4238
+tests/langchain/test_langchain.py,sha256=p1YkeL1U23b0E3vhEcXonwMa3QGUswQjKNReMbV0ndA,2576
+tests/langchain/test_langchain_community.py,sha256=B7Z7YRx7v9507wEyDnXV1NKVYo4r2XU0YplugfYVsJ0,2615
+tests/langchain/test_langchain_core.py,sha256=LBeoK27-uDethSvBGykYAlCVfEwljQrN7sZL91TVCPw,4467
 tests/openai/conftest.py,sha256=BkehS6heg-O91Nzoc4546OSiAzy8KgSgk7VCO3A11zM,700
 tests/openai/test_chat_completion.py,sha256=-TwCAF2czsiGZqETQvMOgmIJgdFWycJ9nOeO5q65AJs,5804
 tests/openai/test_embeddings.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -120,8 +121,8 @@ tests/openai/cassettes/test_async_image_generation.yaml,sha256=_LYZcrqxrnSqcWVQn
 tests/openai/cassettes/test_chat_completion.yaml,sha256=YkNFgK9VHAzNqGWuxFcTiE194GdEie8eDf1FSsffjd8,2944
 tests/openai/cassettes/test_chat_completion_streaming.yaml,sha256=nkx_TemQMYSZxUF_b-LCEFwCRDm0AkQHLf4sdJVuZBw,2592394
 tests/openai/cassettes/test_image_generation.yaml,sha256=gn5aSVp6V6_hb_rt2NnkAWd_idzDxo-7VzhZII0Wslw,3562
-tests/pinecone/test_pinecone.py,sha256=_wlJbSKnY7gyzVcwxIWKft1P_t8dWwcIKNfGCrRLiHs,2633
-langtrace_python_sdk-2.0.4.dist-info/METADATA,sha256=XjCobjWDtvEejEkuyfq_aQebtTgd04WPx-H5BajSDQA,9585
-langtrace_python_sdk-2.0.4.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-langtrace_python_sdk-2.0.4.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-2.0.4.dist-info/RECORD,,
+tests/pinecone/test_pinecone.py,sha256=iFuAqzbfrxHzH_J2kDCbezqYv1m-yeQiHiO3rMKTuYc,2735
+langtrace_python_sdk-2.0.6.dist-info/METADATA,sha256=To2Jd-L4FM7sHtVWbVbSU19s0BQKynZm9LKboKRLrmQ,9582
+langtrace_python_sdk-2.0.6.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+langtrace_python_sdk-2.0.6.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-2.0.6.dist-info/RECORD,,

tests/chroma/test_chroma.py

@@ -1,8 +1,7 @@
 from unittest.mock import MagicMock, patch, call
-from langtrace_python_sdk.constants.instrumentation.common import \
-    SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from opentelemetry.trace.status import StatusCode,Status
+from opentelemetry.trace.status import StatusCode, Status
 from langtrace_python_sdk.instrumentation.chroma.patch import collection_patch
 from opentelemetry.trace import SpanKind
 from tests.utils import common_setup
@@ -14,11 +13,14 @@ class TestChromaPatch(unittest.TestCase):
     data = {
         "status": "success",
     }
+
     def setUp(self):
-            self.chroma_mock, self.tracer, self.span = common_setup(self.data, 'chromadb.Collection.add')
-            self.wrapped_method = MagicMock(return_value="mocked method result")
-            self.instance = MagicMock()
-            self.instance.name = "aa" 
+        self.chroma_mock, self.tracer, self.span = common_setup(
+            self.data, "chromadb.Collection.add"
+        )
+        self.wrapped_method = MagicMock(return_value="mocked method result")
+        self.instance = MagicMock()
+        self.instance.name = "aa"
 
     def tearDown(self):
         self.chroma_mock.stop()
@@ -35,18 +37,22 @@ class TestChromaPatch(unittest.TestCase):
         self.assertEqual(result, "mocked method result")
 
         # Assert the span is started with the correct parameters
-        self.assertTrue(self.tracer.start_as_current_span.called_once_with("chromadb.Collection.add", kind=SpanKind.CLIENT))
+        self.assertTrue(
+            self.tracer.start_as_current_span.called_once_with(
+                "chromadb.Collection.add", kind=SpanKind.CLIENT
+            )
+        )
 
         # Verify span attributes are set as expected
         expected_attributes = {
-            'langtrace.sdk.name': 'langtrace-python-sdk',
-            'langtrace.service.name': 'Chroma',
-            'langtrace.service.type': 'vectordb',
-            'langtrace.service.version': '1.2.3',
-            'langtrace.version': '1.0.0',
-            'db.system': 'chromadb',
-            'db.operation': 'add',
-            'db.collection.name': 'aa',
+            "langtrace.sdk.name": "langtrace-python-sdk",
+            "langtrace.service.name": "Chroma",
+            "langtrace.service.type": "vectordb",
+            "langtrace.service.version": "1.2.3",
+            "langtrace.version": "1.0.0",
+            "db.system": "chromadb",
+            "db.operation": "add",
+            "db.collection.name": "aa",
         }
         for key, value in expected_attributes.items():
             self.span.set_attribute.assert_has_calls([call(key, value)], any_order=True)
@@ -55,10 +61,10 @@ class TestChromaPatch(unittest.TestCase):
 
         for key, value in expected_attributes.items():
             self.assertIn(call(key, value), actual_calls)
-        
+
         # Assert the span status is set to OK
         self.span.set_status.assert_called_with(StatusCode.OK)
-        
 
-if __name__ == '__main__':
-    unittest.main()
+
+if __name__ == "__main__":
+    unittest.main()

tests/langchain/test_langchain.py

@@ -1,4 +1,3 @@
-
 import unittest
 from unittest.mock import MagicMock, call
 from langtrace_python_sdk.instrumentation.langchain.patch import generic_patch
@@ -10,15 +9,17 @@ from opentelemetry.trace.status import Status, StatusCode
 from tests.utils import common_setup
 import json
 
+
 class TestGenericPatch(unittest.TestCase):
     data = {"key": "value"}
+
     def setUp(self):
         self.langchain_mock, self.tracer, self.span = common_setup(self.data, None)
 
     def tearDown(self):
         # Clean up after each test case
         pass
-      
+
     def test_generic_patch(self):
         # Arrange
         method_name = "example_method"
@@ -26,44 +27,56 @@ class TestGenericPatch(unittest.TestCase):
         trace_input = False  # Change as per your requirement
         args = (1, 2, 3)
         task = "split_text"
-        kwargs = {'key': 'value'}
-        version = importlib.metadata.version('langchain')
+        kwargs = {"key": "value"}
+        version = importlib.metadata.version("langchain")
 
         # Act
-        wrapped_function = generic_patch("langchain.text_splitter", task, self.tracer, version, trace_output, trace_input)
+        wrapped_function = generic_patch(
+            "langchain.text_splitter",
+            task,
+            self.tracer,
+            version,
+            trace_output,
+            trace_input,
+        )
         result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
 
         # Assert
-        self.assertTrue(self.tracer.start_as_current_span.called_once_with(method_name, kind=SpanKind.CLIENT))
-        
+        self.assertTrue(
+            self.tracer.start_as_current_span.called_once_with(
+                method_name, kind=SpanKind.CLIENT
+            )
+        )
+
         service_provider = "Langchain"
         expected_attributes = {
-            'langtrace.sdk.name': 'langtrace-python-sdk',
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
         }
-        
-        
+
         self.assertTrue(
             self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()], any_order=True
+                [call(key, value) for key, value in expected_attributes.items()],
+                any_order=True,
             )
         )
-                
+
         actual_calls = self.span.set_attribute.call_args_list
 
         for key, value in expected_attributes.items():
             self.assertIn(call(key, value), actual_calls)
-        
+
         self.assertEqual(self.span.set_status.call_count, 1)
         self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
 
-        expected_result_data = {"key": "value"  }   
+        expected_result_data = {"key": "value"}
 
         self.assertEqual(result.key, expected_result_data["key"])
 
-if __name__ == '__main__':
-    unittest.main()
+
+if __name__ == "__main__":
+    unittest.main()

tests/langchain/test_langchain_community.py

@@ -1,4 +1,3 @@
-
 import unittest
 from unittest.mock import MagicMock, Mock, patch, call
 from langtrace_python_sdk.instrumentation.langchain_community.patch import generic_patch
@@ -10,8 +9,11 @@ from langtrace_python_sdk.constants.instrumentation.openai import APIS
 from opentelemetry.trace.status import Status, StatusCode
 import json
 from tests.utils import common_setup
+
+
 class TestGenericPatch(unittest.TestCase):
     data = {"key": "value"}
+
     def setUp(self):
         self.langchain_mock, self.tracer, self.span = common_setup(self.data, None)
 
@@ -23,47 +25,58 @@ class TestGenericPatch(unittest.TestCase):
         # Arrange
         method_name = "example_method"
         trace_output = False
-        trace_input = False 
+        trace_input = False
         args = (1, 2, 3)
         task = "vector_store"
-        kwargs = {'key': 'value'}
+        kwargs = {"key": "value"}
         version = importlib.metadata.version("langchain-community")
 
         # Act
-        wrapped_function = generic_patch("langchain_community.vectorstores.faiss", task, self.tracer, version, trace_output, trace_input)
+        wrapped_function = generic_patch(
+            "langchain_community.vectorstores.faiss",
+            task,
+            self.tracer,
+            version,
+            trace_output,
+            trace_input,
+        )
         result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
 
         # Assert
-        self.assertTrue(self.tracer.start_as_current_span.called_once_with(method_name, kind=SpanKind.CLIENT))
-        
+        self.assertTrue(
+            self.tracer.start_as_current_span.called_once_with(
+                method_name, kind=SpanKind.CLIENT
+            )
+        )
+
         service_provider = "Langchain Community"
         expected_attributes = {
-            'langtrace.sdk.name': 'langtrace-python-sdk',
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
         }
-        
-        
+
         self.assertTrue(
             self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()], any_order=True
+                [call(key, value) for key, value in expected_attributes.items()],
+                any_order=True,
             )
         )
-                
+
         actual_calls = self.span.set_attribute.call_args_list
 
         for key, value in expected_attributes.items():
             self.assertIn(call(key, value), actual_calls)
-        
+
         self.assertEqual(self.span.set_status.call_count, 1)
         self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
 
-        expected_result_data = {"key": "value"  }   
+        expected_result_data = {"key": "value"}
         self.assertEqual(result.key, expected_result_data["key"])
 
 
-if __name__ == '__main__':
-    unittest.main()
+if __name__ == "__main__":
+    unittest.main()

tests/langchain/test_langchain_core.py

@@ -1,7 +1,9 @@
-
 import unittest
 from unittest.mock import MagicMock, Mock, patch, call
-from langtrace_python_sdk.instrumentation.langchain_core.patch import generic_patch, runnable_patch
+from langtrace_python_sdk.instrumentation.langchain_core.patch import (
+    generic_patch,
+    runnable_patch,
+)
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace import get_tracer
 import importlib.metadata
@@ -10,8 +12,11 @@ from langtrace_python_sdk.constants.instrumentation.openai import APIS
 from opentelemetry.trace.status import Status, StatusCode
 import json
 from tests.utils import common_setup
+
+
 class TestGenericPatch(unittest.TestCase):
     data = {"items": "value"}
+
     def setUp(self):
         self.langchain_mock, self.tracer, self.span = common_setup(self.data, None)
 
@@ -23,67 +28,88 @@ class TestGenericPatch(unittest.TestCase):
         # Arrange
         method_name = "example_method"
         trace_output = False
-        trace_input = True 
+        trace_input = True
         task = "retriever"
         args = (1, 2, 3)
-        kwargs = {'key': 'value'}
+        kwargs = {"key": "value"}
         version = importlib.metadata.version("langchain-core")
 
         # Act
-        wrapped_function = generic_patch("langchain_core.retrievers", task , self.tracer, version, trace_output, trace_input)
+        wrapped_function = generic_patch(
+            "langchain_core.retrievers",
+            task,
+            self.tracer,
+            version,
+            trace_output,
+            trace_input,
+        )
         result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
 
         # Assert
-        self.assertTrue(self.tracer.start_as_current_span.called_once_with(method_name, kind=SpanKind.CLIENT))
-        
+        self.assertTrue(
+            self.tracer.start_as_current_span.called_once_with(
+                method_name, kind=SpanKind.CLIENT
+            )
+        )
+
         service_provider = "Langchain Core"
         expected_attributes = {
-            'langtrace.sdk.name': 'langtrace-python-sdk',
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
         }
-        
+
         self.assertTrue(
             self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()], any_order=True
+                [call(key, value) for key, value in expected_attributes.items()],
+                any_order=True,
             )
         )
-             
+
         actual_calls = self.span.set_attribute.call_args_list
         for key, value in expected_attributes.items():
             self.assertIn(call(key, value), actual_calls)
 
-        
         self.assertEqual(self.span.set_status.call_count, 1)
         self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
 
-        expected_result_data = {"items": "value"  }   
+        expected_result_data = {"items": "value"}
         self.assertEqual(result.items, expected_result_data["items"])
 
     def test_runnable_patch(self):
         # Arrange
         method_name = "example_method"
         trace_output = False
-        trace_input = True 
+        trace_input = True
         args = (1, 2, 3)
-        kwargs = {'key': 'value'}
+        kwargs = {"key": "value"}
         version = importlib.metadata.version("langchain-core")
 
         # Act
-        wrapped_function = runnable_patch("langchain_core.runnables.passthrough",
-                "runnablepassthrough", self.tracer, version, trace_output, trace_input)
+        wrapped_function = runnable_patch(
+            "langchain_core.runnables.passthrough",
+            "runnablepassthrough",
+            self.tracer,
+            version,
+            trace_output,
+            trace_input,
+        )
 
         result = wrapped_function(self.langchain_mock, MagicMock(), args, kwargs)
 
         # Assert
-        self.assertTrue(self.tracer.start_as_current_span.called_once_with(method_name, kind=SpanKind.CLIENT))
-        
+        self.assertTrue(
+            self.tracer.start_as_current_span.called_once_with(
+                method_name, kind=SpanKind.CLIENT
+            )
+        )
+
         service_provider = "Langchain Core"
         expected_attributes = {
-            'langtrace.sdk.name': 'langtrace-python-sdk',
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
@@ -93,23 +119,23 @@ class TestGenericPatch(unittest.TestCase):
 
         self.assertTrue(
             self.span.set_attribute.has_calls(
-                [call(key, value) for key, value in expected_attributes.items()], any_order=True
+                [call(key, value) for key, value in expected_attributes.items()],
+                any_order=True,
             )
         )
-        
+
         actual_calls = self.span.set_attribute.call_args_list
 
         for key, value in expected_attributes.items():
             self.assertIn(call(key, value), actual_calls)
 
-       
         self.assertEqual(self.span.set_status.call_count, 1)
         self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
 
-        expected_result_data = {"items": "value"  }   
+        expected_result_data = {"items": "value"}
 
         self.assertEqual(result.items, expected_result_data["items"])
 
 
-if __name__ == '__main__':
-    unittest.main()
+if __name__ == "__main__":
+    unittest.main()