Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
25 changes: 25 additions & 0 deletions dspy/adapters/xml_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,31 @@ def format_signature_fields_for_instructions(fields: dict[str, FieldInfo]):
parts.append(format_signature_fields_for_instructions(signature.output_fields))
return "\n\n".join(parts).strip()

def format_user_message_content(
self,
signature: type[Signature],
inputs: dict[str, Any],
prefix: str = "",
suffix: str = "",
main_request: bool = False,
) -> str:
messages = [prefix]

messages.append(self.format_field_with_value(
{
FieldInfoWithName(name=k, info=v): inputs.get(k)
for k, v in signature.input_fields.items() if k in inputs
},
))

if main_request:
output_requirements = self.user_message_output_requirements(signature)
if output_requirements is not None:
messages.append(output_requirements)

messages.append(suffix)
return "\n\n".join(messages).strip()

def format_assistant_message_content(
self,
signature: type[Signature],
Expand Down
2 changes: 1 addition & 1 deletion tests/adapters/test_xml_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -291,7 +291,7 @@ class QA(dspy.Signature):
)

expected_user = (
"[[ ## query ## ]]\nwhen was Marie Curie born\n\n"
"<query>\nwhen was Marie Curie born\n</query>\n\n"
"Respond with the corresponding output fields wrapped in XML tags `<answer>`."
)

Expand Down