SDK: embedded server mode (#3949)
This commit is contained in:
committed by
GitHub
parent
6319847709
commit
4ee3a52522
21
skyvern/library/embedded_server_factory.py
Normal file
21
skyvern/library/embedded_server_factory.py
Normal file
@@ -0,0 +1,21 @@
|
||||
import httpx
|
||||
from httpx import ASGITransport
|
||||
|
||||
from skyvern.client import AsyncSkyvern, SkyvernEnvironment
|
||||
from skyvern.config import settings
|
||||
from skyvern.forge.api_app import app
|
||||
|
||||
|
||||
def create_embedded_server(
|
||||
api_key: str,
|
||||
open_api_key: str | None,
|
||||
) -> AsyncSkyvern:
|
||||
if open_api_key:
|
||||
settings.OPENAI_API_KEY = open_api_key
|
||||
|
||||
transport = ASGITransport(app=app)
|
||||
return AsyncSkyvern(
|
||||
environment=SkyvernEnvironment.LOCAL,
|
||||
api_key=api_key,
|
||||
httpx_client=httpx.AsyncClient(transport=transport, base_url="http://skyvern-embedded"),
|
||||
)
|
||||
Reference in New Issue
Block a user