Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/workflows/code_format.yml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ jobs:
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: '3.10'
python-version: '3.11'
cache: 'pip' # caching pip dependencies

- name: Pip install
Expand All @@ -31,4 +31,4 @@ jobs:
run: pip list

- name: Code Formatting
run: black . --check
run: black . --check --diff
7 changes: 7 additions & 0 deletions src/agentlab/agents/generic_agent/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,10 @@
RANDOM_SEARCH_AGENT,
AGENT_4o,
AGENT_4o_MINI,
AGENT_CLAUDE_SONNET_35,
AGENT_4o_VISION,
AGENT_4o_MINI_VISION,
AGENT_CLAUDE_SONNET_35_VISION,
)

__all__ = [
Expand All @@ -28,4 +31,8 @@
"AGENT_8B",
"RANDOM_SEARCH_AGENT",
"AGENT_CUSTOM",
"AGENT_CLAUDE_SONNET_35",
"AGENT_4o_VISION",
"AGENT_4o_MINI_VISION",
"AGENT_CLAUDE_SONNET_35_VISION",
]
14 changes: 14 additions & 0 deletions src/agentlab/agents/generic_agent/agent_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -260,6 +260,10 @@
chat_model_args=CHAT_MODEL_ARGS_DICT["openai/gpt-4o-mini-2024-07-18"],
flags=FLAGS_GPT_4o,
)
AGENT_CLAUDE_SONNET_35 = GenericAgentArgs(
chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/anthropic/claude-3.5-sonnet:beta"],
flags=FLAGS_GPT_4o,
)

# GPT-4o vision default config
FLAGS_GPT_4o_VISION = FLAGS_GPT_4o.copy()
Expand All @@ -271,6 +275,16 @@
flags=FLAGS_GPT_4o_VISION,
)

AGENT_4o_MINI_VISION = GenericAgentArgs(
chat_model_args=CHAT_MODEL_ARGS_DICT["openai/gpt-4o-mini-2024-07-18"],
flags=FLAGS_GPT_4o_VISION,
)

AGENT_CLAUDE_SONNET_35_VISION = GenericAgentArgs(
chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/anthropic/claude-3.5-sonnet:beta"],
flags=FLAGS_GPT_4o_VISION,
)


DEFAULT_RS_FLAGS = GenericPromptFlags(
flag_group="default_rs",
Expand Down
2 changes: 1 addition & 1 deletion src/agentlab/agents/generic_agent/reproducibility_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
This module contains the classes and functions to reproduce the results of a
study. It is used to create a new study that will run the same experiments as
the original study, but with a reproducibility agent that will mimic the same
answers as the original agent.
answers as the original agent.

Stats are collected to compare the original agent's answers with the new agent's
answers. Load the this reproducibility study in agent-xray to compare the results.
Expand Down
8 changes: 6 additions & 2 deletions src/agentlab/llm/llm_configs.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,9 @@
from agentlab.llm.chat_api import (AzureModelArgs, OpenAIModelArgs,
OpenRouterModelArgs, SelfHostedModelArgs)
from agentlab.llm.chat_api import (
AzureModelArgs,
OpenAIModelArgs,
OpenRouterModelArgs,
SelfHostedModelArgs,
)

default_oss_llms_args = {
"n_retry_server": 4,
Expand Down