Merge pull request #121 from ag2ai/asynctesttext

Update test to avoid Azure content filter
ag2ai · Dec 2, 2024 · 7de2f21 · 7de2f21
2 parents 4889c58 + 5994c56
commit 7de2f21
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/agentchat/test_async.py b/test/agentchat/test_async.py
@@ -23,8 +23,8 @@ def get_market_news(ind, ind_upper):
     data = {
         "feed": [
             {
-                "title": "Palantir CEO Says Our Generation's Atomic Bomb Could Be AI Weapon - And Arrive Sooner Than You Think - Palantir Technologies  ( NYSE:PLTR ) ",
-                "summary": "Christopher Nolan's blockbuster movie \"Oppenheimer\" has reignited the public discourse surrounding the United States' use of an atomic bomb on Japan at the end of World War II.",
+                "title": "Palantir CEO Says Our Generation's Main Challenge Could Be AI Against Humanity - And Arrive Sooner Than You Think - Palantir Technologies  ( NYSE:PLTR ) ",
+                "summary": "Christopher Nolan's blockbuster movie \"Oppenheimer\" has reignited the public discourse surrounding the United States' use of a weapon on Japan at the end of World War II.",
                 "overall_sentiment_score": 0.009687,
             },
             {
@@ -63,7 +63,7 @@ def get_market_news(ind, ind_upper):
 @pytest.mark.skipif(skip_openai, reason=reason)
 @pytest.mark.asyncio
 async def test_async_groupchat():
-    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, KEY_LOC, filter_dict={"tags": ["gpt-3.5-turbo"]})
+    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, KEY_LOC, filter_dict={"tags": ["gpt-4o-mini"]})
 
     # create an AssistantAgent instance named "assistant"
     assistant = autogen.AssistantAgent(
@@ -97,7 +97,7 @@ async def test_async_groupchat():
 @pytest.mark.skipif(skip_openai, reason=reason)
 @pytest.mark.asyncio
 async def test_stream():
-    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, KEY_LOC, filter_dict={"tags": ["gpt-3.5-turbo"]})
+    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, KEY_LOC, filter_dict={"tags": ["gpt-4o-mini"]})
     data = asyncio.Future()
 
     async def add_stock_price_data():