Files
agentic_security/agentic_security/__main__.py
T
Alexander Myasoedov 74461efaa0 feat(Integrated Garak):
2024-04-27 21:44:38 +03:00

34 lines
595 B
Python

import os
import sys
import fire
import uvicorn
from agentic_security.app import app
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("."))
def entrypoint():
fire.Fire(T().server)
def ci_entrypoint():
fire.Fire(T().headless)
if __name__ == "__main__":
entrypoint()