Skip to content

Commit 6f4731a

Browse files
feat: Adding demo_agent project with traces (#7391)
* check author * adding demo project traces (datasets and evals not yet) * changing name to demo_agent * cleanup * cleanup * style: ruff auto-format * removing prints * removing empty file * remove evalsfixture comment Co-authored-by: Xander Song <[email protected]> --------- Co-authored-by: Xander Song <[email protected]>
1 parent b582f95 commit 6f4731a

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

src/phoenix/trace/fixtures.py

+24
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,29 @@ class TracesFixture:
139139
),
140140
)
141141

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+
142165
demo_code_based_agent_fixture = TracesFixture(
143166
name="demo_code_based_agent",
144167
project_name="demo_agents",
@@ -298,6 +321,7 @@ class TracesFixture:
298321
vision_fixture,
299322
anthropic_tools_fixture,
300323
project_sessions_llama_index_rag_arize_docs_fixture,
324+
demo_toolcalling_fixture,
301325
]
302326

303327
NAME_TO_TRACES_FIXTURE: dict[str, TracesFixture] = {

0 commit comments

Comments
 (0)