This will fix when tools show up at the wrong areas of the site like …

…chat completion
LlmLaraHub · Aug 13, 2024 · 9fa0023 · 9fa0023
1 parent e95fde5
commit 9fa0023
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 7 deletions.
diff --git a/Modules/LlmDriver/app/BaseClient.php b/Modules/LlmDriver/app/BaseClient.php
@@ -297,12 +297,12 @@ public function remapFunctions(array $functions): array
     /**
      * @DEPRECATED
      * This was before all the LLMs had tools
-     * 
+     *
      * @param  MessageInDto[]  $messages
      */
     protected function insertFunctionsIntoMessageArray(array $messages): array
     {
-        $functions = $this->setToolType(ToolTypes::Source)->getFunctions();
+        $functions = $this->setToolType(ToolTypes::ChatCompletion)->getFunctions();
 
         $functionsEncoded = collect($functions)->transform(
             function ($item) {

diff --git a/Modules/LlmDriver/app/Functions/GatherInfoTool.php b/Modules/LlmDriver/app/Functions/GatherInfoTool.php
@@ -22,7 +22,7 @@ class GatherInfoTool extends FunctionContract
     use ToolsHelper;
 
     public array $toolTypes = [
-        ToolTypes::ChatCompletion,
+        ToolTypes::ManualChoice,
     ];
 
     protected string $name = 'gather_info_tool';

diff --git a/Modules/LlmDriver/app/Functions/GetWebSiteFromUrlTool.php b/Modules/LlmDriver/app/Functions/GetWebSiteFromUrlTool.php
@@ -15,7 +15,7 @@ class GetWebSiteFromUrlTool extends FunctionContract
     public bool $showInUi = false;
 
     public array $toolTypes = [
-        ToolTypes::ChatCompletion,
+        ToolTypes::ManualChoice,
         ToolTypes::Source,
         ToolTypes::Output,
     ];

diff --git a/Modules/LlmDriver/app/Functions/ReportingTool.php b/Modules/LlmDriver/app/Functions/ReportingTool.php
@@ -24,7 +24,7 @@ class ReportingTool extends FunctionContract
     public bool $showInUi = true;
 
     public array $toolTypes = [
-        ToolTypes::ChatCompletion,
+        ToolTypes::ManualChoice,
     ];
 
     protected string $name = 'reporting_tool';

diff --git a/Modules/LlmDriver/app/Functions/SearchTheWeb.php b/Modules/LlmDriver/app/Functions/SearchTheWeb.php
@@ -20,7 +20,7 @@ class SearchTheWeb extends FunctionContract
     public bool $showInUi = false;
 
     public array $toolTypes = [
-        ToolTypes::ChatCompletion,
+        ToolTypes::ManualChoice,
         ToolTypes::Source,
         ToolTypes::Output,
     ];

diff --git a/Modules/LlmDriver/app/Functions/StandardsChecker.php b/Modules/LlmDriver/app/Functions/StandardsChecker.php
@@ -20,7 +20,7 @@ class StandardsChecker extends FunctionContract
     protected array $results = [];
 
     public array $toolTypes = [
-        ToolTypes::ChatCompletion,
+        ToolTypes::ManualChoice,
     ];
 
     protected array $promptHistory = [];

diff --git a/Modules/LlmDriver/app/Functions/ToolTypes.php b/Modules/LlmDriver/app/Functions/ToolTypes.php
@@ -13,4 +13,5 @@ enum ToolTypes: string
     case Source = 'source';
     case Output = 'output';
     case NoFunction = 'no_function';
+    case ManualChoice = 'manual_choice';
 }