init

AlgoriAgent/examples/conversation_basic/conversation.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+"""A simple example for conversation between user and assistant agent."""
+import agentscope
+from agentscope.agents import DialogAgent
+from agentscope.agents.user_agent import UserAgent
+from agentscope.pipelines.functional import sequentialpipeline
+
+
+def main() -> None:
+    """A basic conversation demo"""
+
+    agentscope.init(
+        model_configs=[
+            {
+                "model_type": "openai_chat",
+                "config_name": "gpt-3.5-turbo",
+                "model_name": "gpt-3.5-turbo",
+                "api_key": "xxx",  # Load from env if not provided
+                "organization": "xxx",  # Load from env if not provided
+                "generate_args": {
+                    "temperature": 0.5,
+                },
+            },
+            {
+                "model_type": "post_api_chat",
+                "config_name": "my_post_api",
+                "api_url": "https://xxx",
+                "headers": {},
+            },
+        ],
+        project="Multi-Agent Conversation",
+    )
+
+    # Init two agents
+    dialog_agent = DialogAgent(
+        name="Assistant",
+        sys_prompt="You're a helpful assistant.",
+        model_config_name="gpt-3.5-turbo",  # replace by your model config name
+    )
+    user_agent = UserAgent()
+
+    # start the conversation between user and assistant
+    x = None
+    while x is None or x.content != "exit":
+        x = sequentialpipeline([dialog_agent, user_agent], x)
+
+
+if __name__ == "__main__":
+    main()