Skip to content

Commit 38d0668

Browse files
committed
fix: mypy problem with ragbits prompt output type (#772)
1 parent 7a20886 commit 38d0668

File tree

1 file changed

+9
-0
lines changed
  • packages/ragbits-core/src/ragbits/core/llms

1 file changed

+9
-0
lines changed

packages/ragbits-core/src/ragbits/core/llms/base.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -364,6 +364,15 @@ async def generate_raw(
364364

365365
return returned
366366

367+
@overload
368+
async def generate(
369+
self,
370+
prompt: BasePromptWithParser[PromptOutputT],
371+
*,
372+
tools: None = None,
373+
options: LLMClientOptionsT | None = None,
374+
) -> PromptOutputT: ...
375+
367376
@overload
368377
async def generate(
369378
self,

0 commit comments

Comments
 (0)