Files
agentic_security/agentic_security/__main__.py
T
2025-01-04 19:20:52 +02:00

36 lines
686 B
Python

import os
import sys
import fire
import uvicorn
from agentic_security.app import app
from agentic_security.lib import AgenticSecurity
class T:
def server(self, port=8718, host="0.0.0.0"):
sys.path.append(os.path.dirname("."))
config = uvicorn.Config(
app, port=port, host=host, log_level="info", reload=True
)
server = uvicorn.Server(config)
server.run()
return
def headless(self):
sys.path.append(os.path.dirname("."))
AgenticSecurity().entrypoint()
def entrypoint():
fire.Fire(T().server)
def ci_entrypoint():
fire.Fire(T().headless)
if __name__ == "__main__":
ci_entrypoint()