Add gpt-4o-mini, update examples (#253)

This commit is contained in:
Eric Zhu
2024-07-23 18:05:16 -07:00
committed by GitHub
parent 2288aee72b
commit 176932ef9c
12 changed files with 29 additions and 22 deletions

View File

@@ -255,14 +255,14 @@ async def main() -> None:
"ReviewerAgent",
lambda: ReviewerAgent(
description="Code Reviewer",
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini"),
),
)
await runtime.register(
"CoderAgent",
lambda: CoderAgent(
description="Coder",
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini"),
),
)
run_context = runtime.start()

View File

@@ -118,7 +118,7 @@ async def main() -> None:
lambda: GroupChatParticipant(
description="A data scientist",
system_messages=[SystemMessage("You are a data scientist.")],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini"),
),
)
agent2 = await runtime.register_and_get(
@@ -126,7 +126,7 @@ async def main() -> None:
lambda: GroupChatParticipant(
description="An engineer",
system_messages=[SystemMessage("You are an engineer.")],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini"),
),
)
agent3 = await runtime.register_and_get(
@@ -134,7 +134,7 @@ async def main() -> None:
lambda: GroupChatParticipant(
description="An artist",
system_messages=[SystemMessage("You are an artist.")],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini"),
),
)

View File

@@ -117,7 +117,7 @@ async def main() -> None:
lambda: ReferenceAgent(
description="Reference Agent 1",
system_messages=[SystemMessage("You are a helpful assistant that can answer questions.")],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo", temperature=0.1),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini", temperature=0.1),
),
)
await runtime.register(
@@ -125,7 +125,7 @@ async def main() -> None:
lambda: ReferenceAgent(
description="Reference Agent 2",
system_messages=[SystemMessage("You are a helpful assistant that can answer questions.")],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo", temperature=0.5),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini", temperature=0.5),
),
)
await runtime.register(
@@ -133,7 +133,7 @@ async def main() -> None:
lambda: ReferenceAgent(
description="Reference Agent 3",
system_messages=[SystemMessage("You are a helpful assistant that can answer questions.")],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo", temperature=1.0),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini", temperature=1.0),
),
)
await runtime.register(
@@ -145,7 +145,7 @@ async def main() -> None:
"...synthesize these responses into a single, high-quality response... Responses from models:"
)
],
model_client=get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
model_client=get_chat_completion_client_from_envs(model="gpt-4o-mini"),
num_references=3,
),
)

View File

@@ -214,7 +214,7 @@ async def main(question: str) -> None:
await runtime.register(
"MathSolver1",
lambda: MathSolver(
get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
get_chat_completion_client_from_envs(model="gpt-4o-mini"),
neighbor_names=["MathSolver2", "MathSolver4"],
max_round=3,
),
@@ -222,7 +222,7 @@ async def main(question: str) -> None:
await runtime.register(
"MathSolver2",
lambda: MathSolver(
get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
get_chat_completion_client_from_envs(model="gpt-4o-mini"),
neighbor_names=["MathSolver1", "MathSolver3"],
max_round=3,
),
@@ -230,7 +230,7 @@ async def main(question: str) -> None:
await runtime.register(
"MathSolver3",
lambda: MathSolver(
get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
get_chat_completion_client_from_envs(model="gpt-4o-mini"),
neighbor_names=["MathSolver2", "MathSolver4"],
max_round=3,
),
@@ -238,7 +238,7 @@ async def main(question: str) -> None:
await runtime.register(
"MathSolver4",
lambda: MathSolver(
get_chat_completion_client_from_envs(model="gpt-3.5-turbo"),
get_chat_completion_client_from_envs(model="gpt-4o-mini"),
neighbor_names=["MathSolver1", "MathSolver3"],
max_round=3,
),