fixed tests

gptme/codeblock.py

@@ -1,5 +1,5 @@
-from dataclasses import dataclass
 from collections.abc import Generator
+from dataclasses import dataclass
 from xml.etree import ElementTree
 
 

gptme/reduce.py

@@ -8,7 +8,7 @@
 from collections.abc import Generator
 from copy import copy
 
-from ..codeblock import Codeblock
+from .codeblock import Codeblock
 from .message import Message, len_tokens
 from .models import get_model
 

gptme/tools/base.py

@@ -38,9 +38,7 @@ def get_doc(self, _doc="") -> str:
             _doc += "\n\n"
         _doc += f"ToolSpec: {self.name}\n{self.desc}"
         if self.instructions:
-            _doc += f"\n# Instructions: {self.instructions}"
+            _doc += f"\n\n# Instructions\n\n{self.instructions}"
         if self.examples:
-            _doc += (
-                f"\n# Examples: {transform_examples_to_chat_directives(self.examples)}"
-            )
+            _doc += f"\n\n# Examples\n\n{transform_examples_to_chat_directives(self.examples)}"
         return _doc

gptme/tools/patch.py

@@ -1,40 +1,8 @@
 """
 Gives the LLM agent the ability to patch text files, by using a adapted version git conflict markers.
-
-The format is suitable for small changes to files we have written in the past, without having to rewrite the whole file.
-
-
-Format:
-
-.. code-block:: patch
-
-    ```patch <filename>
-    <<<<<<< ORIGINAL
-    original code
-    =======
-    modified code
-    >>>>>>> UPDATED
-    ```
-
-Example:
-
-.. chat::
-
-    User: edit `hello.py` to ask for the name to greet
-    Assistant:
-    ```patch hello.py
-    <<<<<<< ORIGINAL
-    print("Hello world")
-    =======
-    name = input("What is your name? ")
-    print(f"Hello {name}")
-    >>>>>>> UPDATED
-    ```
-    System: Patch applied
 """
 
 # TODO: support multiple patches in one codeblock (or make it clear that only one patch per codeblock is supported/applied)
-
 import re
 from collections.abc import Generator
 from pathlib import Path
@@ -90,6 +58,7 @@ def patch_to_output(patch: str, filename: str, append: bool = False) -> str:
     print(f"Hello {name}")
 >>>>>>> UPDATED
 ''')}
+> System: Patch applied
 
 > User: change the codeblock to append to the file
 > Assistant: {patch_to_output("patch.py", '''

tests/test_codeblock.py

@@ -11,7 +11,7 @@ def hello():
 More text
 """
     assert Codeblock.iter_from_markdown(markdown) == [
-        ("python", 'def hello():\n    print("Hello, World!")')
+        Codeblock("python", 'def hello():\n    print("Hello, World!")')
     ]
 
 
@@ -31,11 +31,11 @@ def greet(name):
 ```
 """
     assert Codeblock.iter_from_markdown(markdown) == [
-        (
+        Codeblock(
             "java",
             'public class Main {\n    public static void main(String[] args) {\n        System.out.println("Hello, Java!");\n    }\n}',
         ),
-        ("python", 'def greet(name):\n    return f"Hello, {name}!"'),
+        Codeblock("python", 'def greet(name):\n    return f"Hello, {name}!"'),
     ]
 
 
@@ -51,7 +51,7 @@ def print_readme():
 ```
 """
     assert Codeblock.iter_from_markdown(markdown) == [
-        (
+        Codeblock(
             "python",
             "def print_readme():\n    print('''Usage:\n```javascript\ncallme()\n```\n''')",
         )
@@ -77,5 +77,5 @@ def hello():
 ```
 """
     assert Codeblock.iter_from_markdown(markdown) == [
-        ("", 'def hello():\n    print("Hello, World!")')
+        Codeblock("", 'def hello():\n    print("Hello, World!")')
     ]

tests/test_logmanager.py

@@ -1,3 +1,4 @@
+from gptme.codeblock import Codeblock
 from gptme.logmanager import LogManager, Message
 
 
@@ -18,7 +19,7 @@ def test_get_last_codeblock():
 """,
         )
     )
-    assert ("python", "print('world')") == log.get_last_codeblock()
+    assert Codeblock("python", "print('world')") == log.get_last_codeblock()
 
 
 def test_branch():