feat: refactor startup process and instrument httpx

pyproject.toml

@@ -21,7 +21,7 @@ dependencies = [
     "zhipuai~=2.1.0.20240521",
     "anthropic~=0.28.0",
     "dashscope~=1.19.2",
-    "logfire[fastapi,system-metrics]~=0.42.0",
+    "logfire[fastapi,system-metrics,httpx]~=0.42.0",
 ]
 
 [tool.pdm]

src/__init__.py

@@ -0,0 +1,25 @@
+from contextlib import suppress
+
+with suppress(ModuleNotFoundError):
+    from dotenv import load_dotenv
+
+    load_dotenv(override=True)
+
+from .utils.config import env
+
+if env.logfire_token:
+    import logfire
+
+    logfire.configure()
+    logfire.info("app started", **env.model_dump())
+    logfire.instrument_httpx()
+    logfire.instrument_openai()
+    logfire.instrument_anthropic()
+
+    from .entry import app
+
+    logfire.instrument_fastapi(app)
+
+from .utils.load import generate_pyi
+
+generate_pyi()

src/entry.py

@@ -1,16 +1,13 @@
-from dotenv import load_dotenv
 from fastapi import FastAPI
 from fastapi.responses import PlainTextResponse
 from starlette.middleware.cors import CORSMiddleware
 from starlette.staticfiles import StaticFiles
 
 from .routes.prompts import prompts_router
 from .routes.run import run_router
-from .utils.config import env
-from .utils.load import generate_pyi
 from .utils.time import now
 
-app = FastAPI(title="Promplate Demo", description="<https://github.com/promplate/demo>", on_startup=[generate_pyi, load_dotenv])
+app = FastAPI(title="Promplate Demo", description="<https://github.com/promplate/demo>")
 app.add_middleware(CORSMiddleware, allow_origins="*", allow_credentials=True, allow_methods="*", allow_headers="*")
 app.include_router(prompts_router, prefix="/prompts")
 app.include_router(run_router)
@@ -22,12 +19,3 @@ async def greet():
 
 
 app.mount("/", StaticFiles(directory="frontend/dist", html=True, check_dir=False))
-
-if env.logfire_token:
-    import logfire
-
-    logfire.configure()
-    logfire.info("app started", **env.model_dump())
-    logfire.instrument_openai()
-    logfire.instrument_anthropic()
-    logfire.instrument_fastapi(app)