Skip to content

Make Teleprompt.compile more typing friendly. #9013

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
+245 −141
Open

Make Teleprompt.compile more typing friendly. #9013

Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
55aef6f
Added a replaced teleprompt.compile's signature with a TypeVar that t…
TyTodd Nov 3, 2025
315840f
added typevar to all other compilers
TyTodd Nov 3, 2025
e45875e
cleanup
TyTodd Nov 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
Added a replaced teleprompt.compile's signature with a TypeVar that t… 
…racks the type exact type of the student module passed in
  • Loading branch information
@TyTodd
TyTodd committed Nov 3, 2025
commit 55aef6f0d29528fb6ba1a6f5e338808d57403e6a
14 changes: 12 additions & 2 deletions dspy/teleprompt/teleprompt.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,13 +1,23 @@
from typing import Any
from typing import Any, TypeVar

from dspy.primitives import Example, Module

M = TypeVar("M", bound=Module)


class Teleprompter:
def __init__(self):
pass

def compile(self, student: Module, *, trainset: list[Example], teacher: Module | None = None, valset: list[Example] | None = None, **kwargs) -> Module:
def compile(
self,
student: M,
*,
trainset: list[Example],
teacher: Module | None = None,
valset: list[Example] | None = None,
**kwargs,
) -> M:
"""
Optimize the student program.

Expand Down