fix tests

LlmLaraHub · Jul 4, 2024 · 081813b · 081813b
1 parent 4358279
commit 081813b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Modules/LlmDriver/app/Orchestrate.php b/Modules/LlmDriver/app/Orchestrate.php
@@ -65,14 +65,15 @@ public function handle(
                  * @TODO
                  * Refactor this since Message really can build this
                  * and I am now passing this into all things.
+                 * Will come back shortly
                  */
                 $functionDto = FunctionCallDto::from([
                     'arguments' => '{}',
                     'function_name' => 'standards_checker',
                     'filter' => $filter,
                 ]);
 
-                $response = StandardsChecker::handle($messagesArray, $chat, $functionDto);
+                $response = StandardsChecker::handle($message);
                 $messagesArray = $this->handleResponse($response, $chat);
                 $this->response = $response->content;
                 $this->requiresFollowup = $response->requires_follow_up_prompt;

diff --git a/Modules/LlmDriver/app/Requests/MessageInDto.php b/Modules/LlmDriver/app/Requests/MessageInDto.php
@@ -21,7 +21,7 @@ public function toArray(): array
     {
         return [
             'content' => $this->content,
-            'role' => $this->role->value,
+            'role' => $this->role,
         ];
     }