Skip to content

Commit 46eaf8c

Browse files
authored
fix(vertexai): support generative model chat session tracing (#2689)
1 parent 394a340 commit 46eaf8c

File tree

3 files changed

+643
-81
lines changed

3 files changed

+643
-81
lines changed

packages/opentelemetry-instrumentation-vertexai/opentelemetry/instrumentation/vertexai/__init__.py

+14
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,13 @@
4141
"span_name": "vertexai.generate_content_async",
4242
"is_async": True,
4343
},
44+
{
45+
"package": "vertexai.generative_models",
46+
"object": "ChatSession",
47+
"method": "send_message",
48+
"span_name": "vertexai.send_message",
49+
"is_async": False,
50+
},
4451
{
4552
"package": "vertexai.preview.generative_models",
4653
"object": "GenerativeModel",
@@ -55,6 +62,13 @@
5562
"span_name": "vertexai.generate_content_async",
5663
"is_async": True,
5764
},
65+
{
66+
"package": "vertexai.preview.generative_models",
67+
"object": "ChatSession",
68+
"method": "send_message",
69+
"span_name": "vertexai.send_message",
70+
"is_async": False,
71+
},
5872
{
5973
"package": "vertexai.language_models",
6074
"object": "TextGenerationModel",

0 commit comments

Comments
 (0)