- additional prompt tweaks;

plasma-umass · Aug 26, 2024 · 9b84650 · 9b84650
1 parent a239fd6
commit 9b84650
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/coverup/prompt.py b/src/coverup/prompt.py
@@ -161,7 +161,7 @@ def get_info(ctx: CodeSegment, name: str) -> str:
         """
         {
             "name": "get_info",
-            "description": "Returns information about a function, class or method.",
+            "description": "Returns information about a symbol.",
             "parameters": {
                 "type": "object",
                 "properties": {
@@ -178,9 +178,9 @@ def get_info(ctx: CodeSegment, name: str) -> str:
         from .codeinfo import get_info, parse_file
 
         if info := get_info(parse_file(ctx.path), name):
-            return info
+            return "\"...\" below indicates omitted code.\n\n" + info
 
-        return "Unable to obtain information."
+        return f"Unable to obtain information on {name}."
 
 
     def get_functions(self) -> T.List[T.Callable]: