(
prompt: str,
model_type: str,
model_id: str,
provider: str | None = None,
api_base: str | None = None,
api_key: str | None = None,
)
| 215 | |
| 216 | |
| 217 | def run_webagent( |
| 218 | prompt: str, |
| 219 | model_type: str, |
| 220 | model_id: str, |
| 221 | provider: str | None = None, |
| 222 | api_base: str | None = None, |
| 223 | api_key: str | None = None, |
| 224 | ) -> None: |
| 225 | # Load environment variables |
| 226 | load_dotenv() |
| 227 | |
| 228 | # Initialize the model based on the provided arguments |
| 229 | model = load_model(model_type, model_id, provider=provider, api_base=api_base, api_key=api_key) |
| 230 | |
| 231 | global driver |
| 232 | driver = initialize_driver() |
| 233 | agent = initialize_agent(model) |
| 234 | |
| 235 | # Run the agent with the provided prompt |
| 236 | agent.python_executor("from helium import *") |
| 237 | agent.run(prompt + helium_instructions) |
| 238 | |
| 239 | |
| 240 | def main() -> None: |
searching dependent graphs…