Skip to content

Commit c6f06b9

Browse files
chore(internal): improve examples
1 parent 41397b2 commit c6f06b9

File tree

2 files changed

+70
-10
lines changed

2 files changed

+70
-10
lines changed

tests/api_resources/agents/test_query.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,13 @@ def test_method_create_with_all_params(self, client: ContextualAI) -> None:
4949
retrievals_only=True,
5050
conversation_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
5151
documents_filters={
52-
"filters": [],
52+
"filters": [
53+
{
54+
"field": "field1",
55+
"operator": "equals",
56+
"value": "value1",
57+
}
58+
],
5359
"operator": "AND",
5460
},
5561
llm_model_id="llm_model_id",
@@ -322,7 +328,13 @@ async def test_method_create_with_all_params(self, async_client: AsyncContextual
322328
retrievals_only=True,
323329
conversation_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
324330
documents_filters={
325-
"filters": [],
331+
"filters": [
332+
{
333+
"field": "field1",
334+
"operator": "equals",
335+
"value": "value1",
336+
}
337+
],
326338
"operator": "AND",
327339
},
328340
llm_model_id="llm_model_id",

tests/api_resources/test_agents.py

Lines changed: 56 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,24 @@ def test_method_create_with_all_params(self, client: ContextualAI) -> None:
3636
agent_configs={
3737
"filter_and_rerank_config": {
3838
"default_metadata_filters": {
39-
"filters": [],
39+
"filters": [
40+
{
41+
"field": "field1",
42+
"operator": "equals",
43+
"value": "value1",
44+
}
45+
],
4046
"operator": "AND",
4147
},
4248
"per_datastore_metadata_filters": {
4349
"d49609d9-61c3-4a67-b3bd-5196b10da560": {
44-
"filters": [],
50+
"filters": [
51+
{
52+
"field": "field1",
53+
"operator": "equals",
54+
"value": "value1",
55+
}
56+
],
4557
"operator": "AND",
4658
}
4759
},
@@ -124,12 +136,24 @@ def test_method_update_with_all_params(self, client: ContextualAI) -> None:
124136
agent_configs={
125137
"filter_and_rerank_config": {
126138
"default_metadata_filters": {
127-
"filters": [],
139+
"filters": [
140+
{
141+
"field": "field1",
142+
"operator": "equals",
143+
"value": "value1",
144+
}
145+
],
128146
"operator": "AND",
129147
},
130148
"per_datastore_metadata_filters": {
131149
"d49609d9-61c3-4a67-b3bd-5196b10da560": {
132-
"filters": [],
150+
"filters": [
151+
{
152+
"field": "field1",
153+
"operator": "equals",
154+
"value": "value1",
155+
}
156+
],
133157
"operator": "AND",
134158
}
135159
},
@@ -410,12 +434,24 @@ async def test_method_create_with_all_params(self, async_client: AsyncContextual
410434
agent_configs={
411435
"filter_and_rerank_config": {
412436
"default_metadata_filters": {
413-
"filters": [],
437+
"filters": [
438+
{
439+
"field": "field1",
440+
"operator": "equals",
441+
"value": "value1",
442+
}
443+
],
414444
"operator": "AND",
415445
},
416446
"per_datastore_metadata_filters": {
417447
"d49609d9-61c3-4a67-b3bd-5196b10da560": {
418-
"filters": [],
448+
"filters": [
449+
{
450+
"field": "field1",
451+
"operator": "equals",
452+
"value": "value1",
453+
}
454+
],
419455
"operator": "AND",
420456
}
421457
},
@@ -498,12 +534,24 @@ async def test_method_update_with_all_params(self, async_client: AsyncContextual
498534
agent_configs={
499535
"filter_and_rerank_config": {
500536
"default_metadata_filters": {
501-
"filters": [],
537+
"filters": [
538+
{
539+
"field": "field1",
540+
"operator": "equals",
541+
"value": "value1",
542+
}
543+
],
502544
"operator": "AND",
503545
},
504546
"per_datastore_metadata_filters": {
505547
"d49609d9-61c3-4a67-b3bd-5196b10da560": {
506-
"filters": [],
548+
"filters": [
549+
{
550+
"field": "field1",
551+
"operator": "equals",
552+
"value": "value1",
553+
}
554+
],
507555
"operator": "AND",
508556
}
509557
},

0 commit comments

Comments
 (0)