From e010383fd1cd0a6ec81d1f9051c42abf11f698b2 Mon Sep 17 00:00:00 2001 From: jung0072 Date: Sun, 12 May 2024 11:01:58 -0400 Subject: [PATCH] fix bug on build_signature method in langchain --- dspy/predict/langchain.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dspy/predict/langchain.py b/dspy/predict/langchain.py index 86d439f50c..6592ddd07d 100644 --- a/dspy/predict/langchain.py +++ b/dspy/predict/langchain.py @@ -9,7 +9,7 @@ from dspy.predict.parameter import Parameter from dspy.predict.predict import Predict from dspy.primitives.prediction import Prediction -from dspy.signatures.field import InputField, OutputField +from dspy.signatures.field import OldInputField, OldOutputField from dspy.signatures.signature import infer_prefix # TODO: This class is currently hard to test, because it hardcodes gpt-4 usage: @@ -73,8 +73,8 @@ def _build_signature(self, template): with dspy.context(lm=gpt4T): parts = dspy.Predict(Template2Signature)(template=template) - inputs = {k.strip(): InputField() for k in parts.input_keys.split(',')} - outputs = {k.strip(): OutputField() for k in parts.output_key.split(',')} + inputs = {k.strip(): OldInputField() for k in parts.input_keys.split(',')} + outputs = {k.strip(): OldOutputField() for k in parts.output_key.split(',')} for k, v in inputs.items(): v.finalize(k, infer_prefix(k)) # TODO: Generate from the template at dspy.Predict(Template2Signature)