Implemented function calling for user stories

2026-01-21 17:48:05 +01:00 · 2023-07-28 15:09:57 +02:00
parent 32e988db24
commit 81e4e92a08
5 changed files with 50 additions and 9 deletions
--- a/euclid/const/function_calls.py
+++ b/euclid/const/function_calls.py
@@ -0,0 +1,33 @@
+from utils.llm import parse_llm_output
+
+def process_user_stories(stories):
+    return stories
+
+def return_array_from_prompt(values_in_list):
+    return {
+        'name': 'process_user_stories',
+        'description': f"Print the list of user stories that are created.",
+        'parameters': {
+            'type': 'object',
+            "properties": {
+                "stories": {
+                    "type": "array",
+                    "description": f"List of user stories that are created in a list.",
+                    "items": {
+                        "type": "string",
+                        "description": "User story"
+                    },
+                },
+            },
+            "required": ["stories"],
+        },
+    }
+
+USER_STORIES = {
+    'definitions': [
+        return_array_from_prompt('user stories')
+    ],
+    'functions': {
+        'process_user_stories': process_user_stories
+    },
+}