@@ -139,6 +139,29 @@ class TracesFixture:
139
139
),
140
140
)
141
141
142
+ demo_toolcalling_fixture = TracesFixture (
143
+ name = "demo_toolcalling" ,
144
+ project_name = "demo_agent" ,
145
+ description = "Tool calling traces" ,
146
+ file_name = "agents-toolcalling-tracesv2.parquet" ,
147
+ dataset_fixtures = (
148
+ DatasetFixture (
149
+ file_name = "questions.csv.gz" ,
150
+ input_keys = ("query" ,),
151
+ output_keys = ("responses" ,),
152
+ name = "Valid Queries" ,
153
+ description = "Valid queries for the demo agent" ,
154
+ ),
155
+ DatasetFixture (
156
+ file_name = "invalid_questions.csv.gz" ,
157
+ input_keys = ("query" ,),
158
+ output_keys = ("responses" ,),
159
+ name = "Invalid Queries" ,
160
+ description = "Invalid queries for the demo agent" ,
161
+ ),
162
+ ),
163
+ )
164
+
142
165
demo_code_based_agent_fixture = TracesFixture (
143
166
name = "demo_code_based_agent" ,
144
167
project_name = "demo_agents" ,
@@ -298,6 +321,7 @@ class TracesFixture:
298
321
vision_fixture ,
299
322
anthropic_tools_fixture ,
300
323
project_sessions_llama_index_rag_arize_docs_fixture ,
324
+ demo_toolcalling_fixture ,
301
325
]
302
326
303
327
NAME_TO_TRACES_FIXTURE : dict [str , TracesFixture ] = {
0 commit comments