weave LLMCaller.llm_key through to api handler/agent (#2524)

This commit is contained in:
Shuchang Zheng
2025-05-29 13:49:59 -07:00
committed by GitHub
parent 7f6b65ba61
commit ea5620acd2
3 changed files with 34 additions and 18 deletions

View File

@@ -94,6 +94,7 @@ class LLMAPIHandler(Protocol):
ai_suggestion: AISuggestion | None = None,
screenshots: list[bytes] | None = None,
parameters: dict[str, Any] | None = None,
llm_key_override: str | None = None,
) -> Awaitable[dict[str, Any]]: ...
@@ -106,5 +107,6 @@ async def dummy_llm_api_handler(
ai_suggestion: AISuggestion | None = None,
screenshots: list[bytes] | None = None,
parameters: dict[str, Any] | None = None,
llm_key_override: str | None = None,
) -> dict[str, Any]:
raise NotImplementedError("Your LLM provider is not configured. Please configure it in the .env file.")