diff --git a/src/documentation/fastapi_example.py b/src/documentation/fastapi_example.py index 8459fed8..cbf001ff 100644 --- a/src/documentation/fastapi_example.py +++ b/src/documentation/fastapi_example.py @@ -65,7 +65,7 @@ def __call__( def client() -> Client: return Client( token=os.environ["AA_TOKEN"], - host=os.getenv("AA_CLIENT_BASE_URL", "https://api.aleph-alpha.com"), + host=os.environ["CLIENT_URL"], ) @@ -78,7 +78,7 @@ def default_model( def summary_task( model: Annotated[LuminousControlModel, Depends(default_model)], ) -> SteerableSingleChunkSummarize: - return SteerableSingleChunkSummarize(model) + return SteerableSingleChunkSummarize(model=model) @app.post(