mirror of
https://github.com/langchain-ai/langchain-extract.git
synced 2026-07-01 20:24:03 -04:00
Eugene Yurtsev
26 lines
679 B
Python
26 lines
679 B
Python
from contextlib import asynccontextmanager
|
|
from typing import Optional
|
|
|
|
import httpx
|
|
from fastapi import FastAPI
|
|
from httpx import AsyncClient
|
|
|
|
|
|
@asynccontextmanager
|
|
async def get_async_test_client(
|
|
server: FastAPI, *, path: Optional[str] = None, raise_app_exceptions: bool = True
|
|
) -> AsyncClient:
|
|
"""Get an async client."""
|
|
url = "http://localhost:9999/"
|
|
if path:
|
|
url += path
|
|
transport = httpx.ASGITransport(
|
|
app=server,
|
|
raise_app_exceptions=raise_app_exceptions,
|
|
)
|
|
async_client = AsyncClient(app=server, base_url=url, transport=transport)
|
|
try:
|
|
yield async_client
|
|
finally:
|
|
await async_client.aclose()
|