add tool type

LlmLaraHub · Sep 22, 2024 · a1b032e · a1b032e
1 parent 52065db
commit a1b032e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app/Domains/Sources/WebhookSource.php b/app/Domains/Sources/WebhookSource.php
@@ -16,6 +16,7 @@
 use Illuminate\Support\Arr;
 use Illuminate\Support\Facades\Bus;
 use Illuminate\Support\Facades\Log;
+use LlmLaraHub\LlmDriver\Functions\ToolTypes;
 use LlmLaraHub\LlmDriver\LlmDriverFacade;
 use LlmLaraHub\TagFunction\Jobs\TagDocumentJob;
 
@@ -64,7 +65,8 @@ public function handle(Source $source): void
 
         $results = LlmDriverFacade::driver(
             $source->getDriver()
-        )->completion($prompt);
+        )->setToolType(ToolTypes::Source)
+            ->completion($prompt);
 
         if ($this->ifNotActionRequired($results->content)) {
             Log::info('[LaraChain] - Webhook Skipping', [