this should be a first stab at open ai functions now for claude

LlmLaraHub · Apr 5, 2024 · dbf4468 · dbf4468
1 parent 6b148b8
commit dbf4468
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 6 deletions.
diff --git a/tests/Feature/Http/Controllers/ChatControllerTest.php b/tests/Feature/Http/Controllers/ChatControllerTest.php
@@ -4,7 +4,6 @@
 
 use App\Models\Chat;
 use App\Models\Collection;
-use App\Models\Message;
 use App\Models\User;
 use Facades\App\LlmDriver\Orchestrate;
 use Tests\TestCase;
@@ -67,9 +66,7 @@ public function test_kick_off_chat_makes_system()
                 'system_prompt' => 'Foo',
                 'input' => 'user input',
             ])->assertOk();
-        $this->assertDatabaseCount('messages', 4);
+        $this->assertDatabaseCount('messages', 3);
 
-        $this->assertTrue(Message::whereRole('system')->exists());
-        $this->assertTrue(Message::where('is_chat_ignored', true)->exists());
     }
 }
diff --git a/tests/Feature/MockClientTest.php b/tests/Feature/MockClientTest.php
@@ -19,7 +19,7 @@ public function test_tools(): void
 
         $this->assertCount(1, $results);
 
-        $this->assertEquals('search_and_summarize', $results[0]['name']);
+        $this->assertEquals('summarize_collection', $results[0]['name']);
     }
 
     public function test_tool_with_limit(): void
@@ -28,7 +28,7 @@ public function test_tool_with_limit(): void
 
         $results = $client->functionPromptChat(['test'], ['search_and_summarize']);
 
-        $this->assertCount(0, $results);
+        $this->assertCount(1, $results);
     }
 
     public function test_embeddings(): void

diff --git a/tests/fixtures/openai_client_get_functions.json b/tests/fixtures/openai_client_get_functions.json
@@ -4,6 +4,25 @@
         "function": {
             "name": "search_and_summarize",
             "description": "Used to embed users prompt, search database and return summarized results.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "prompt": {
+                        "description": "This is the prompt the user is using to search the database and may or may not assist the results.",
+                        "type": "string",
+                        "enum": [],
+                        "default": ""
+                    }
+                }
+            },
+            "required": []
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "summarize_collection",
+            "description": "This is used when the prompt wants to summarize the entire collection of documents",
             "parameters": {
                 "type": "object",
                 "properties": {