github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Remove dependency on RetrieveAssistantAgent for RetrieveChat (#3320)

Browse Source 
* Remove deps on RetrieveAssistantAgent for getting human input

* Terminate when no more context

* Add deprecation warning message

* Clean up RetrieveAssistantAgent, part 1

* Update version

* Clean up docs and notebooks
This commit is contained in:
Li Jiang
2024-08-16 00:03:06 +08:00
committed by GitHub
parent 736d5e72bc
commit 08fa1b6d08
17 changed files with 567 additions and 479 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/agentchat/contrib/retrievechat/test_pgvector_retrievechat.py
View File

@@ -6,8 +6,7 @@ import sys
import pytest
from sentence_transformers import SentenceTransformer
from autogen import config_list_from_json
from autogen.agentchat.contrib.retrieve_assistant_agent import RetrieveAssistantAgent
from autogen import AssistantAgent, config_list_from_json
sys.path.append(os.path.join(os.path.dirname(__file__), "../../.."))
from conftest import skip_openai # noqa: E402
@@ -18,9 +17,6 @@ from test_assistant_agent import KEY_LOC, OAI_CONFIG_LIST # noqa: E402
try:
import pgvector
from autogen.agentchat.contrib.retrieve_assistant_agent import (
RetrieveAssistantAgent,
)
from autogen.agentchat.contrib.retrieve_user_proxy_agent import (
RetrieveUserProxyAgent,
)
@@ -46,7 +42,7 @@ def test_retrievechat():
file_location=KEY_LOC,
)
assistant = RetrieveAssistantAgent(
assistant = AssistantAgent(
name="assistant",
system_message="You are a helpful assistant.",
llm_config={

5
test/agentchat/contrib/retrievechat/test_qdrant_retrievechat.py
View File

@@ -5,8 +5,7 @@ import sys
import pytest
from autogen import config_list_from_json
from autogen.agentchat.contrib.retrieve_assistant_agent import RetrieveAssistantAgent
from autogen import AssistantAgent, config_list_from_json
sys.path.append(os.path.join(os.path.dirname(__file__), "../../.."))
from conftest import skip_openai # noqa: E402
@@ -51,7 +50,7 @@ def test_retrievechat():
file_location=KEY_LOC,
)
assistant = RetrieveAssistantAgent(
assistant = AssistantAgent(
name="assistant",
system_message="You are a helpful assistant.",
llm_config={

6
test/agentchat/contrib/retrievechat/test_retrievechat.py
View File

@@ -18,9 +18,7 @@ try:
import openai
from chromadb.utils import embedding_functions as ef
from autogen.agentchat.contrib.retrieve_assistant_agent import (
RetrieveAssistantAgent,
)
from autogen import AssistantAgent
from autogen.agentchat.contrib.retrieve_user_proxy_agent import (
RetrieveUserProxyAgent,
)
@@ -45,7 +43,7 @@ def test_retrievechat():
file_location=KEY_LOC,
)
assistant = RetrieveAssistantAgent(
assistant = AssistantAgent(
name="assistant",
system_message="You are a helpful assistant.",
llm_config={