18 lines
586 B
Python
Executable File
18 lines
586 B
Python
Executable File
from typing import Dict, Optional
|
|
from mm_agents.os_symphony.core.mllm import LMMAgent
|
|
|
|
|
|
class BaseModule:
|
|
def __init__(self, engine_params: Dict = None, platform: str = "Linux"):
|
|
self.engine_params = engine_params
|
|
self.platform = platform
|
|
|
|
def _create_agent(
|
|
self, system_prompt: str = None, engine_params: Optional[Dict] = None
|
|
) -> LMMAgent:
|
|
"""Create a new LMMAgent instance"""
|
|
agent = LMMAgent(engine_params or self.engine_params)
|
|
if system_prompt:
|
|
agent.add_system_prompt(system_prompt)
|
|
return agent
|