increment

Author: chenmoneygithub

dspy/adapters/types/base_type.py

@@ -70,6 +70,11 @@ def serialize_model(self):
             )
         return formatted
 
+    @classmethod
+    def is_natively_supported(cls, lm, lm_kwargs) -> bool:
+        """Whether the custom type is natively supported by the LM."""
+        return False
+
     @classmethod
     def is_streamable(cls) -> bool:
         """Whether the custom type is streamable."""
@@ -88,7 +93,6 @@ def parse_stream_chunk(cls, chunk: ModelResponseStream) -> Optional["Type"]:
         """
         return None
 
-
     @classmethod
     def parse_lm_response(cls, response: str | dict[str, Any]) -> Optional["Type"]:
         """Parse a LM response into the custom type.
@@ -101,6 +105,7 @@ def parse_lm_response(cls, response: str | dict[str, Any]) -> Optional["Type"]:
         """
         return None
 
+
 def split_message_content_for_custom_types(messages: list[dict[str, Any]]) -> list[dict[str, Any]]:
     """Split user message content into a list of content blocks.
 

dspy/adapters/types/reasoning.py

@@ -61,3 +61,12 @@ def parse_lm_response(cls, response: str | dict[str, Any]) -> Optional["Reasonin
 
     def __repr__(self) -> str:
         return f"{self.content!r}"
+
+    def __str__(self) -> str:
+        return self.content
+
+    def __eq__(self, other: object) -> bool:
+        if isinstance(other, Reasoning):
+            return self.content == other.content
+        if isinstance(other, str):
+            return self.content == other

tests/adapters/test_citation.py

@@ -155,7 +155,8 @@ class CitationSignature(Signature):
         CitationSignature.delete("citations"),
         CitationSignature,
         outputs,
-        dspy.LM(model="claude-3-5-sonnet-20241022")
+        dspy.LM(model="claude-3-5-sonnet-20241022"),
+        lm_kwargs={},
     )
 
     assert len(result) == 1

tests/streaming/test_streaming.py

@@ -907,6 +907,7 @@ class CustomSignature(dspy.Signature):
     )
 
     async def stream(*args, **kwargs):
+        yield ModelResponseStream(model="gpt-4o-mini", choices=[StreamingChoices(delta=Delta(content="[[ ## answer ## ]]\n"))])
         yield ModelResponseStream(model="gpt-4o-mini", choices=[StreamingChoices(delta=Delta(content="Hello"))])
         yield ModelResponseStream(model="gpt-4o-mini", choices=[StreamingChoices(delta=Delta(content="World"))])
         yield ModelResponseStream(model="gpt-4o-mini", choices=[StreamingChoices(delta=Delta(content="\n\n"))])