I am trying to run autogen using my own server (runpod), but I am getting the following error:
TypeError: Completions.create() got an unexpected keyword argument 'api_type'
It seems to me that the error is coming from openai, but I am not sure.
I'm using pyautogen~=0.2.0b5
Here is my code:
config_list = [{
"model": "llama-7B",
"base_url": "https://xxxxxxx.proxy.runpod.net/v1",
"api_type": "open_ai",
"api_key": "sk-111111111111111111111111111111111111111111111111"
}]
llm_config = {
"seed": 42,
"config_list": config_list,
"temperature": 0
}
assistant = autogen.AssistantAgent(
name="assistant",
llm_config=llm_config,
is_termination_msg=lambda x: True if "TERMINATE" in x.get(
"content") else False,
)
user_proxy = autogen.UserProxyAgent(
name="user_proxy",
human_input_mode="NEVER",
is_termination_msg=lambda x: True if "TERMINATE" in x.get(
"content") else False,
max_consecutive_auto_reply=10,
code_execution_config={
"work_dir": "work_dir",
"use_docker": False,
},
)
task1 = """
Find arxiv papers that show how are people studying trust calibration in AI based systems
"""
user_proxy.initiate_chat(assistant, message=task1)
I don't know why this is happening. Is anyone else having this issue?