Skip to content

Commit e213d89

Browse files
taskingaijctaskingaijc
committed
test: fix test
1 parent 05eab79 commit e213d89

File tree

2 files changed

+18
-18
lines changed

2 files changed

+18
-18
lines changed

‎test/testcase/test_async/test_async_assistant.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ async def test_a_create_assistant(self):
2020
"model_id": Config.openai_chat_completion_model_id,
2121
"name": "test",
2222
"description": "test for assistant",
23-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
23+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
2424
"system_prompt_template": [
2525
"You know the meaning of various numbers.",
2626
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -96,7 +96,7 @@ async def test_a_update_assistant(self):
9696
{
9797
"name": "openai",
9898
"description": "test for openai",
99-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
99+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
100100
"retrievals": [
101101
RetrievalRef(
102102
type=RetrievalType.COLLECTION,
@@ -328,7 +328,7 @@ async def test_a_generate_message_by_stream(self):
328328
"model_id": Config.openai_chat_completion_model_id,
329329
"name": "test",
330330
"description": "test for assistant",
331-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
331+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
332332
"system_prompt_template": [
333333
"You know the meaning of various numbers.",
334334
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -392,7 +392,7 @@ async def test_a_assistant_by_user_message_retrieval_and_stream(self):
392392
"model_id": Config.openai_chat_completion_model_id,
393393
"name": "test",
394394
"description": "test for assistant",
395-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
395+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
396396
"system_prompt_template": [
397397
"You know the meaning of various numbers.",
398398
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -440,7 +440,7 @@ async def test_a_assistant_by_memory_retrieval_and_stream(self):
440440
"model_id": Config.openai_chat_completion_model_id,
441441
"name": "test",
442442
"description": "test for assistant",
443-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
443+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
444444
"system_prompt_template": [
445445
"You know the meaning of various numbers.",
446446
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -488,7 +488,7 @@ async def test_a_assistant_by_function_call_retrieval_and_stream(self):
488488
"model_id": Config.openai_chat_completion_model_id,
489489
"name": "test",
490490
"description": "test for assistant",
491-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
491+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
492492
"system_prompt_template": [
493493
"You know the meaning of various numbers.",
494494
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -536,7 +536,7 @@ async def test_a_assistant_by_not_support_function_call_retrieval_and_stream(sel
536536
"model_id": Config.anthropic_chat_completion_model_id,
537537
"name": "test",
538538
"description": "test for assistant",
539-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
539+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
540540
"system_prompt_template": [
541541
"You know the meaning of various numbers.",
542542
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -567,7 +567,7 @@ async def test_a_assistant_by_all_tool_and_stream(self):
567567
"model_id": Config.openai_chat_completion_model_id,
568568
"name": "test",
569569
"description": "test for assistant",
570-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
570+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
571571
"system_prompt_template": [
572572
"You know the meaning of various numbers.",
573573
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -614,7 +614,7 @@ async def test_a_assistant_by_not_support_function_call_tool_and_stream(self):
614614
"model_id": Config.anthropic_chat_completion_model_id,
615615
"name": "test",
616616
"description": "test for assistant",
617-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
617+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
618618
"system_prompt_template": [
619619
"You know the meaning of various numbers.",
620620
"No matter what the user's language is, you will use the {{langugae}} to explain.",

‎test/testcase/test_sync/test_sync_assistant.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def test_create_assistant(self):
1919
"model_id": Config.openai_chat_completion_model_id,
2020
"name": "test",
2121
"description": "test for assistant",
22-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
22+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
2323
"system_prompt_template": [
2424
"You know the meaning of various numbers.",
2525
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -92,7 +92,7 @@ def test_update_assistant(self, assistant_id):
9292
{
9393
"name": "openai",
9494
"description": "test for openai",
95-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
95+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
9696
"retrievals": [
9797
RetrievalRef(
9898
type=RetrievalType.COLLECTION,
@@ -301,7 +301,7 @@ def test_generate_message_by_stream(self):
301301
"model_id": Config.openai_chat_completion_model_id,
302302
"name": "test",
303303
"description": "test for assistant",
304-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
304+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
305305
"system_prompt_template": [
306306
"You know the meaning of various numbers.",
307307
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -366,7 +366,7 @@ def test_assistant_by_user_message_retrieval_and_stream(self):
366366
"model_id": Config.openai_chat_completion_model_id,
367367
"name": "test",
368368
"description": "test for assistant",
369-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
369+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
370370
"system_prompt_template": [
371371
"You know the meaning of various numbers.",
372372
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -413,7 +413,7 @@ def test_assistant_by_memory_retrieval_and_stream(self):
413413
"model_id": Config.openai_chat_completion_model_id,
414414
"name": "test",
415415
"description": "test for assistant",
416-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
416+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
417417
"system_prompt_template": [
418418
"You know the meaning of various numbers.",
419419
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -460,7 +460,7 @@ def test_assistant_by_function_call_retrieval_and_stream(self):
460460
"model_id": Config.openai_chat_completion_model_id,
461461
"name": "test",
462462
"description": "test for assistant",
463-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
463+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
464464
"system_prompt_template": [
465465
"You know the meaning of various numbers.",
466466
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -507,7 +507,7 @@ def test_assistant_by_not_support_function_call_retrieval_and_stream(self):
507507
"model_id": Config.anthropic_chat_completion_model_id,
508508
"name": "test",
509509
"description": "test for assistant",
510-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
510+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
511511
"system_prompt_template": [
512512
"You know the meaning of various numbers.",
513513
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -537,7 +537,7 @@ def test_assistant_by_all_tool_and_stream(self):
537537
"model_id": Config.openai_chat_completion_model_id,
538538
"name": "test",
539539
"description": "test for assistant",
540-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
540+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
541541
"system_prompt_template": [
542542
"You know the meaning of various numbers.",
543543
"No matter what the user's language is, you will use the {{langugae}} to explain.",
@@ -583,7 +583,7 @@ def test_assistant_by_not_support_function_call_tool_and_stream(self):
583583
"model_id": Config.anthropic_chat_completion_model_id,
584584
"name": "test",
585585
"description": "test for assistant",
586-
"memory": AssistantMessageWindowMemory(max_messages=50, max_tokens=2000),
586+
"memory": AssistantMessageWindowMemory(max_tokens=2000),
587587
"system_prompt_template": [
588588
"You know the meaning of various numbers.",
589589
"No matter what the user's language is, you will use the {{langugae}} to explain.",

0 commit comments

Comments
 (0)