dspy_quickstart_signature_and_predict_for_question_answering.py

1import dspy
2
3# 1. Setup the Language Model (LM)
4# You can use OpenAI, Anthropic, Ollama, etc. 
5# Here we use the dspy.LM wrapper (requires OPENAI_API_KEY in environment)
6lm = dspy.LM('openai/gpt-4o-mini')
7dspy.configure(lm=lm)
8
9# 2. Define a Signature
10# Signatures describe the task: they specify inputs and outputs.
11class SimpleQA(dspy.Signature):
12    """Answer questions with short factoid answers."""
13    question = dspy.InputField()
14    answer = dspy.OutputField(desc="often between 1 and 5 words")
15
16# 3. Define the Module
17# Here we use dspy.Predict, which is the most basic module.
18# You can also use dspy.ChainOfThought for better reasoning.
19qa = dspy.Predict(SimpleQA)
20
21# 4. Call the Module
22question = "What is the capital of France?"
23response = qa(question=question)
24
25# 5. Print the Output
26print(f"Question: {question}")
27print(f"Answer: {response.answer}")