mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-30 02:27:01 +00:00
d50f654748
* update orm branch + accesibility tweaks * general file location refactor * add support for LocalCommandLineCodeExecutor and DockerCommandLineCodeExecutor * update code execution config laoding * version update * bump version rc1 * add model type selection (openai , gemini, azure) * add ability to test workflow * psycopg3 support * add close logic to build tab pop ups, enable testing of workflows in build view * updates to dbmanager, version bump * add max_tokens default value * ensure sessions are used correctly in dbmanager * initial support for migrations * update sessions/workflow api routing for clarity. * general refactor, + add support for initial sample workflows * orm branch updates * Removed incorrect Git LFS files * update git lfs tracking --------- Co-authored-by: Audel Rouhi <knucklessg1@gmail.com>
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import os
|
|
from typing import Optional
|
|
|
|
import typer
|
|
import uvicorn
|
|
from typing_extensions import Annotated
|
|
|
|
from .version import VERSION
|
|
|
|
app = typer.Typer()
|
|
|
|
|
|
@app.command()
|
|
def ui(
|
|
host: str = "127.0.0.1",
|
|
port: int = 8081,
|
|
workers: int = 1,
|
|
reload: Annotated[bool, typer.Option("--reload")] = False,
|
|
docs: bool = False,
|
|
appdir: str = None,
|
|
database_uri: Optional[str] = None,
|
|
):
|
|
"""
|
|
Run the AutoGen Studio UI.
|
|
|
|
Args:
|
|
host (str, optional): Host to run the UI on. Defaults to 127.0.0.1 (localhost).
|
|
port (int, optional): Port to run the UI on. Defaults to 8081.
|
|
workers (int, optional): Number of workers to run the UI with. Defaults to 1.
|
|
reload (bool, optional): Whether to reload the UI on code changes. Defaults to False.
|
|
docs (bool, optional): Whether to generate API docs. Defaults to False.
|
|
appdir (str, optional): Path to the AutoGen Studio app directory. Defaults to None.
|
|
database-uri (str, optional): Database URI to connect to. Defaults to None. Examples include sqlite:///autogenstudio.db, postgresql://user:password@localhost/autogenstudio.
|
|
"""
|
|
|
|
os.environ["AUTOGENSTUDIO_API_DOCS"] = str(docs)
|
|
if appdir:
|
|
os.environ["AUTOGENSTUDIO_APPDIR"] = appdir
|
|
if database_uri:
|
|
os.environ["AUTOGENSTUDIO_DATABASE_URI"] = database_uri
|
|
|
|
uvicorn.run(
|
|
"autogenstudio.web.app:app",
|
|
host=host,
|
|
port=port,
|
|
workers=workers,
|
|
reload=reload,
|
|
)
|
|
|
|
|
|
@app.command()
|
|
def version():
|
|
"""
|
|
Print the version of the AutoGen Studio UI CLI.
|
|
"""
|
|
|
|
typer.echo(f"AutoGen Studio CLI version: {VERSION}")
|
|
|
|
|
|
def run():
|
|
app()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|