SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Merge pull request #622 from Joannall/master

Browse source 
Improve Primary Planning
This commit is contained in:
Haiping 2024-09-04 16:01:36 -05:00 committed by GitHub
parent 1a42d91192 348888ae63
commit 35cbee58ff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 43 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
src/Plugins/BotSharp.Plugin.Planner/Functions/PrimaryStagePlanFn.cs
View file

@ -1,4 +1,8 @@
using BotSharp.Abstraction.Conversations.Models;
using BotSharp.Abstraction.Functions;
using BotSharp.Abstraction.Knowledges;
using BotSharp.Abstraction.Knowledges.Models;
using BotSharp.Abstraction.Routing;
using BotSharp.Plugin.Planner.TwoStaging.Models;
namespace BotSharp.Plugin.Planner.Functions;
@ -27,18 +31,26 @@ public class PrimaryStagePlanFn : IFunctionCallback
state.SetState("max_tokens", "4096");
var task = JsonSerializer.Deserialize<PrimaryRequirementRequest>(message.FunctionArgs);
// Get knowledge from vectordb
var collectionName = knowledgeSettings.Default.CollectionName ?? KnowledgeCollectionName.BotSharp; ;
var knowledges = await knowledgeService.SearchVectorKnowledge(task.Requirements, collectionName, new VectorSearchOptions
//get knowledge from vectordb
var knowledges = new List<string>();
foreach (var question in task.Questions)
{
Confidence = 0.1f
});
message.Content = string.Join("\r\n\r\n=====\r\n", knowledges.Select(x => x.ToQuestionAnswer()));
var retrievalMessage = new RoleDialogModel(AgentRole.User, question)
{
FunctionArgs = JsonSerializer.Serialize(new ExtractedKnowledge
{
Question = question
}),
Content = ""
};
await fn.InvokeFunction("knowledge_retrieval", retrievalMessage);
knowledges.Add(retrievalMessage.Content);
}
// Send knowledge to AI to refine and summarize the primary planning
var currentAgent = await agentService.LoadAgent(message.CurrentAgentId);
var firstPlanningPrompt = await GetFirstStagePlanPrompt(task, message);
var firstPlanningPrompt = await GetFirstStagePlanPrompt(task, knowledges);
var plannerAgent = new Agent
{
Id = BuiltInAgentId.Planner,
@ -53,7 +65,7 @@ public class PrimaryStagePlanFn : IFunctionCallback
return true;
}
private async Task<string> GetFirstStagePlanPrompt(PrimaryRequirementRequest task, RoleDialogModel message)
private async Task<string> GetFirstStagePlanPrompt(PrimaryRequirementRequest task, List<string> relevantKnowledges)
{
var agentService = _services.GetRequiredService<IAgentService>();
var render = _services.GetRequiredService<ITemplateRender>();
@ -78,7 +90,7 @@ public class PrimaryStagePlanFn : IFunctionCallback
{
{ "task_description", task.Requirements },
{ "global_knowledges", globalKnowledges },
{ "relevant_knowledges", new[]{ message.Content } },
{ "relevant_knowledges", relevantKnowledges },
{ "response_format", responseFormat }
});
}

7
src/Plugins/BotSharp.Plugin.Planner/TwoStaging/Models/PrimaryRequirementRequest.cs
View file

@ -5,9 +5,6 @@ public class PrimaryRequirementRequest
[JsonPropertyName("requirement_detail")]
public string Requirements { get; set; } = null!;
[JsonPropertyName("has_knowledge_reference")]
public bool HasKnowledgeReference { get; set; }
[JsonPropertyName("question")]
public string Question { get; set; } = null!;
[JsonPropertyName("questions")]
public string[] Questions { get; set; } = [];
}

10
src/Plugins/BotSharp.Plugin.Planner/data/agents/282a7128-69a1-44b0-878c-a9159b88f3b9/instructions/instruction.liquid
View file

@ -1,3 +1,11 @@
Use the TwoStagePlanner approach to plan the overall implementation steps, call plan_primary_stage.
If need_additional_information is true, call plan_secondary_stage for the specific primary stage.
You must Call plan_summary as the last step to summarize the final planning steps.
You must Call plan_summary as the last step to summarize the final planning steps.
{% if global_knowledges != empty -%}
=====
Global Knowledge:
{% for k in global_knowledges %}
{{ k }}
{% endfor %}
{%- endif %}

16
src/Plugins/BotSharp.Plugin.Planner/data/agents/6745151e-6d46-4a02-8de4-1c4f21c7da95/functions/plan_primary_stage.json
View file

@ -1,6 +1,6 @@
{
"name": "plan_primary_stage",
"description": "Extract user's original requirements with detail specifications to make the primary plan, you have to include every important informations.",
"description": "Plan the high level steps to finish the task",
"parameters": {
"type": "object",
"properties": {
@ -8,11 +8,15 @@
"type": "string",
"description": "User original requirements in detail, don't miss any information especially for those line items, values and numbers."
},
"question": {
"type": "string",
"description": "Question convert from requirement and reference tables for knowledge search. The question should contain all the detailed information in the requirement"
"questions": {
"type": "array",
"description": "User requirements in detail, don't miss any information especially for those line items, values and numbers.",
"items": {
"type": "string",
"description": "Question converted from requirement in different ways to search in the knowledge base, be short"
}
}
},
"required": [ "requirement_detail", "question" ]
"required": [ "requirement_detail", "questions" ]
}
}
}