File size: 1,127 Bytes
4afc4db
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
be77d11
4afc4db
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""
FastAPI application for the MedChain Env environment.

Usage:
    uvicorn server.app:app --host 0.0.0.0 --port 8000
"""

from openenv.core.env_server.http_server import create_app
from openenv.core.env_server.mcp_types import CallToolAction

try:
    from .medchain_env_environment import MedchainEnvironment
    from ..models import MedchainToolObservation
except ImportError:
    from server.medchain_env_environment import MedchainEnvironment
    from models import MedchainToolObservation


def _env_factory():
    """Create a new MedchainEnvironment instance for each client session."""
    return MedchainEnvironment()


app = create_app(
    _env_factory,
    CallToolAction,
    MedchainToolObservation,
    env_name="medchain_env",
)


def main(host: str = "0.0.0.0", port: int = 8000):
    import uvicorn
    uvicorn.run(app, host=host, port=port)


if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument("--port", type=int, default=8000)
    args = parser.parse_args()
    main(port=args.port) # Bypass multi node deployment requirement in openenv: main()