25 lines
739 B
Python
25 lines
739 B
Python
from .agent_core import AgentCore
|
|
from .llm_engine import LlmEngine
|
|
from .docker_module import DockerModule
|
|
|
|
def main():
|
|
"""Main entry point for the SIA application."""
|
|
system_prompt_path = "system_prompt.txt"
|
|
action_schema_path = "action_schema.xsd"
|
|
model_path = "/root/model"
|
|
|
|
with open(system_prompt_path, 'r') as f:
|
|
system_prompt = f.read()
|
|
with open(action_schema_path, 'r') as f:
|
|
action_schema = f.read()
|
|
|
|
agent_core = AgentCore(
|
|
system_prompt=f"{system_prompt}{action_schema}",
|
|
action_schema=action_schema,
|
|
docker_module=DockerModule(),
|
|
llm_engine=LlmEngine(model_path=model_path)
|
|
)
|
|
agent_core.run_iteration()
|
|
|
|
if __name__ == "__main__":
|
|
main() |