Skip to content

Commit 6270e95

Browse files
authored
fix mlflow deploy doc example (#8275)
1 parent 7c68c08 commit 6270e95

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/docs/tutorials/deployment/index.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,8 +170,8 @@ class MyProgram(dspy.Module):
170170
super().__init__()
171171
self.cot = dspy.ChainOfThought("question -> answer")
172172

173-
def forward(self, question):
174-
return self.cot(question=question)
173+
def forward(self, messages):
174+
return self.cot(question=messages[0]["content"])
175175

176176
dspy_program = MyProgram()
177177

0 commit comments

Comments
 (0)