ups

tryAGI · Jan 16, 2024 · c6fa507 · c6fa507
1 parent 6918268
commit c6fa507
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/libs/LangChain.Core/Chains/StackableChains/Agents/Crew/Tools/AskQuestionTool.cs b/src/libs/LangChain.Core/Chains/StackableChains/Agents/Crew/Tools/AskQuestionTool.cs
@@ -16,7 +16,7 @@ The input to this tool should be a pipe (|) separated text of length
 For example, `coworker|question|context`.";
     }
 
-    public override string ToolAction(string input)
+    public override Task<string> ToolTask(string input)
     {
         var split = input.Split('|');
         var agent = split[0];
@@ -29,7 +29,7 @@ public override string ToolAction(string input)
                     | coworker;
         var res = chain.Run("result").Result;
 
-        return res;
+        return Task.FromResult(res);
 
     }
 }
diff --git a/src/libs/LangChain.Core/Chains/StackableChains/Agents/Crew/Tools/CrewAgentToolLambda.cs b/src/libs/LangChain.Core/Chains/StackableChains/Agents/Crew/Tools/CrewAgentToolLambda.cs
@@ -9,8 +9,8 @@ public CrewAgentToolLambda(string name, string description, Func<string, string>
         _func = func;
     }
 
-    public override string ToolAction(string input)
+    public override Task<string> ToolTask(string input)
     {
-        return _func(input);
+        return Task.FromResult(_func(input));
     }
 }
diff --git a/src/libs/LangChain.Core/Chains/StackableChains/Agents/Crew/Tools/DelegateWorkTool.cs b/src/libs/LangChain.Core/Chains/StackableChains/Agents/Crew/Tools/DelegateWorkTool.cs
@@ -16,7 +16,7 @@ The input to this tool should be a pipe (|) separated text of length
 For example, `coworker|task|context`";
     }
 
-    public override string ToolAction(string input)
+    public override Task<string> ToolTask(string input)
     {
         var split = input.Split('|');
         var agent = split[0];
@@ -29,6 +29,6 @@ public override string ToolAction(string input)
                     | coworker;
         var res = chain.Run("result").Result;
 
-        return res;
+        return Task.FromResult(res);
     }
 }