Skip to content

Commit 05b3557

Browse files
Merge pull request stanfordnlp#1015 from minki-j/main
Langchain integration bug fixed
2 parents 0e595a7 + e010383 commit 05b3557

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dspy/predict/langchain.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from dspy.predict.parameter import Parameter
1010
from dspy.predict.predict import Predict
1111
from dspy.primitives.prediction import Prediction
12-
from dspy.signatures.field import InputField, OutputField
12+
from dspy.signatures.field import OldInputField, OldOutputField
1313
from dspy.signatures.signature import infer_prefix
1414

1515
# TODO: This class is currently hard to test, because it hardcodes gpt-4 usage:
@@ -73,8 +73,8 @@ def _build_signature(self, template):
7373

7474
with dspy.context(lm=gpt4T): parts = dspy.Predict(Template2Signature)(template=template)
7575

76-
inputs = {k.strip(): InputField() for k in parts.input_keys.split(',')}
77-
outputs = {k.strip(): OutputField() for k in parts.output_key.split(',')}
76+
inputs = {k.strip(): OldInputField() for k in parts.input_keys.split(',')}
77+
outputs = {k.strip(): OldOutputField() for k in parts.output_key.split(',')}
7878

7979
for k, v in inputs.items():
8080
v.finalize(k, infer_prefix(k)) # TODO: Generate from the template at dspy.Predict(Template2Signature)

0 commit comments

Comments
 (0)