diff --git a/Szmedi.RvKits/RvScript/ChatDialogueViewModel.cs b/Szmedi.RvKits/RvScript/ChatDialogueViewModel.cs index 24e3091..c3fcdf0 100644 --- a/Szmedi.RvKits/RvScript/ChatDialogueViewModel.cs +++ b/Szmedi.RvKits/RvScript/ChatDialogueViewModel.cs @@ -1,17 +1,11 @@ -using System; -using System.Collections.Generic; -using System.Collections.ObjectModel; -using System.ComponentModel; +using System.Collections.ObjectModel; using System.Diagnostics; -using System.Linq; using System.Text.RegularExpressions; using System.Threading; using System.Threading.Tasks; using System.Windows; using System.Windows.Controls; -using CommunityToolkit.Mvvm.ComponentModel; -using CommunityToolkit.Mvvm.Input; using CommunityToolkit.Mvvm.Messaging; @@ -21,15 +15,8 @@ using CommunityToolkit.Mvvm.Messaging; using LangChain.Providers; using LangChain.Providers.DeepSeek; using LangChain.Providers.DeepSeek.Predefined; -using LangChain.Providers.OpenAI.Predefined; using Markdig; - -using Markdig.Syntax; -using Markdig.Wpf.ColorCode; - - -using tryAGI.OpenAI; //using LangChain.Schema; //using static LangChain.Chains.Chain; @@ -142,13 +129,13 @@ public partial class ChatDialogueViewModel : ObservableObject { if (ChatHistory.Count == 0) { - CurrentRequest += Message.Human($"你是专业的Revit二次开发工程师，不随意捏造事实，能客观地回答用户的问题，对于不确定或难以理解的问题，需要用户补充说明的，需要主动提出。你需要在{GlobalVariables.UIApplication.Application.VersionName}二次开发中，使用已定义的uidoc、doc两个变量，构造一个可以正确执行的C#代码块，保证使用的RevitAPI的所有方法都能在{GlobalVariables.UIApplication.Application.VersionName}版本的API文档中找到，在代码中添加关键的注释，不需要Excute方法签名和using命名空间，使用类声明对象时，需要完整的命名空间。需求是：{UserInput}"); - ChatHistory.Add(CurrentRequest); + var systemPrompt = new PromptManager(Path.Combine(GlobalVariables.DirAssembly, "Templates/SystemPrompt.txt")).GetSystemPrompt(); + CurrentRequest += Message.Ai($"{systemPrompt}\n\r注意当前的Revit版本是：{GlobalVariables.UIApplication.Application.VersionName}"); + ChatHistory.Add(Message.Human($"{UserInput}")); } else { ChatHistory.Add(Message.Human(UserInput)); - CurrentRequest += Message.Human(UserInput); } diff --git a/Szmedi.RvKits/RvScript/ColorCodeBlockRenderer.cs b/Szmedi.RvKits/RvScript/ColorCodeBlockRenderer.cs index 1400c5d..816a3a3 100644 --- a/Szmedi.RvKits/RvScript/ColorCodeBlockRenderer.cs +++ b/Szmedi.RvKits/RvScript/ColorCodeBlockRenderer.cs @@ -4,14 +4,13 @@ using System.Windows.Documents; using ColorCode; using ColorCode.Styling; -using ColorCode.Wpf; using Markdig.Parsers; using Markdig.Renderers; using Markdig.Renderers.Wpf; using Markdig.Syntax; -#nullable enable -namespace Markdig.Wpf.ColorCode; +using Markdig.Wpf; +namespace Szmedi.RvKits.RvScript; public class ColorCodeBlockRenderer : WpfObjectRenderer { @@ -54,7 +53,7 @@ public class ColorCodeBlockRenderer : WpfObjectRenderer } var code = ExtractCode(codeBlock); - var formatter = new RichTextBoxFormatter(_styleDictionary); + var formatter = new RvScript.RichTextBoxFormatter(_styleDictionary); var paragraph = new Paragraph(); paragraph.SetResourceReference(FrameworkContentElement.StyleProperty, Styles.CodeBlockStyleKey); formatter.FormatInlines(code, language, paragraph.Inlines); @@ -62,7 +61,7 @@ public class ColorCodeBlockRenderer : WpfObjectRenderer renderer.WriteBlock(paragraph); } - private static ILanguage? ExtractLanguage(IFencedBlock fencedCodeBlock, FencedCodeBlockParser parser) + private static ILanguage ExtractLanguage(IFencedBlock fencedCodeBlock, FencedCodeBlockParser parser) { var languageId = fencedCodeBlock.Info!.Replace(parser.InfoPrefix!, string.Empty); diff --git a/Szmedi.RvKits/RvScript/ColorCodeWpfExtension.cs b/Szmedi.RvKits/RvScript/ColorCodeWpfExtension.cs index 25a6a5f..d400bb7 100644 --- a/Szmedi.RvKits/RvScript/ColorCodeWpfExtension.cs +++ b/Szmedi.RvKits/RvScript/ColorCodeWpfExtension.cs @@ -1,8 +1,10 @@ using ColorCode.Styling; + +using Markdig; using Markdig.Renderers; using Markdig.Renderers.Wpf; -namespace Markdig.Wpf.ColorCode; +namespace Szmedi.RvKits.RvScript; public class ColorCodeWpfExtension : IMarkdownExtension { diff --git a/Szmedi.RvKits/RvScript/ExtensionMethods.cs b/Szmedi.RvKits/RvScript/ExtensionMethods.cs index 9c59ff4..d20b2bf 100644 --- a/Szmedi.RvKits/RvScript/ExtensionMethods.cs +++ b/Szmedi.RvKits/RvScript/ExtensionMethods.cs @@ -1,6 +1,6 @@ using System.Windows.Media; -namespace ColorCode.Wpf.Common +namespace Szmedi.RvKits.RvScript { public static class ExtensionMethods { diff --git a/Szmedi.RvKits/RvScript/MarkdownPipelineBuilderExtension.cs b/Szmedi.RvKits/RvScript/MarkdownPipelineBuilderExtension.cs index 1ebaab7..f8d2a5b 100644 --- a/Szmedi.RvKits/RvScript/MarkdownPipelineBuilderExtension.cs +++ b/Szmedi.RvKits/RvScript/MarkdownPipelineBuilderExtension.cs @@ -1,6 +1,8 @@ using ColorCode.Styling; + +using Markdig; #nullable enable -namespace Markdig.Wpf.ColorCode; +namespace Szmedi.RvKits.RvScript; /// /// Extensions for adding ColorCode to the Markdig pipeline. @@ -17,7 +19,7 @@ public static class MarkdownPipelineBuilderExtension this MarkdownPipelineBuilder pipeline, StyleDictionary? styleDictionary = null) { - pipeline.Extensions.Add(new ColorCodeWpfExtension(styleDictionary ?? StyleDictionary.DefaultLight)); + pipeline.Extensions.Add(new RvScript.ColorCodeWpfExtension(styleDictionary ?? StyleDictionary.DefaultLight)); return pipeline; } diff --git a/Szmedi.RvKits/RvScript/PromptManager.cs b/Szmedi.RvKits/RvScript/PromptManager.cs new file mode 100644 index 0000000..6d17857 --- /dev/null +++ b/Szmedi.RvKits/RvScript/PromptManager.cs @@ -0,0 +1,22 @@ +namespace Szmedi.RvKits.RvScript +{ + public class PromptManager + { + private readonly string _systemPrompt; + + public PromptManager(string promptPath) + { + _systemPrompt = File.ReadAllText(promptPath); + } + + public string GetSystemPrompt() + { + return _systemPrompt; + } + + public string BuildUserPrompt(string task) + { + return task.Trim(); + } + } +} diff --git a/Szmedi.RvKits/RvScript/RichTextBoxFormatter.cs b/Szmedi.RvKits/RvScript/RichTextBoxFormatter.cs index 9cd9a65..26b4ee0 100644 --- a/Szmedi.RvKits/RvScript/RichTextBoxFormatter.cs +++ b/Szmedi.RvKits/RvScript/RichTextBoxFormatter.cs @@ -2,12 +2,12 @@ using System.Windows.Controls; using System.Windows.Documents; +using ColorCode; using ColorCode.Common; using ColorCode.Parsing; using ColorCode.Styling; -using ColorCode.Wpf.Common; #nullable enable -namespace ColorCode.Wpf +namespace Szmedi.RvKits.RvScript { /// /// Creates a , for rendering Syntax Highlighted code to a RichTextBlock. @@ -108,7 +108,7 @@ namespace ColorCode.Wpf if (Styles.Contains(scope.Name)) { - Styling.Style style = Styles[scope.Name]; + ColorCode.Styling.Style style = Styles[scope.Name]; foreground = style.Foreground; background = style.Background; diff --git a/Szmedi.RvKits/Szmedi.RvKits.csproj b/Szmedi.RvKits/Szmedi.RvKits.csproj index b99f53e..992fb14 100644 --- a/Szmedi.RvKits/Szmedi.RvKits.csproj +++ b/Szmedi.RvKits/Szmedi.RvKits.csproj @@ -125,7 +125,7 @@ - + diff --git a/Szmedi.RvKits/Templates/SystemPrompt.txt b/Szmedi.RvKits/Templates/SystemPrompt.txt new file mode 100644 index 0000000..818860d --- /dev/null +++ b/Szmedi.RvKits/Templates/SystemPrompt.txt @@ -0,0 +1,250 @@ +你是一名具备大型 BIM 插件架构经验的 Revit API C# 资深工程师。 +你生成的代码将直接进入生产环境，不允许解释性输出或格式污染。 + +--- + +# 一、运行上下文（固定） + +已存在以下变量，且可直接使用： + +```csharp +Autodesk.Revit.DB.Document doc; +Autodesk.Revit.UI.UIDocument uidoc; +``` + +不得重新声明，不得覆盖。 + +--- + +# 二、输出格式（绝对强制） + +你必须： + +1. 仅输出方法体内部代码。 +2. 不得输出： + + * 方法签名 + * 类定义 + * using + * 命名空间 + * XML 注释 + * 解释说明 + * Markdown 说明文字 + * 多余空行 +3. 不得输出任何解释性文本。 +4. 不得在代码块外输出任何字符。 +5. 代码必须包裹在： + +```csharp +``` + +中。 + +6. 代码块内首字符必须是有效 C# 语句。 +7. 代码块结束后不得存在任何字符。 +8. 不允许使用注释（包括 // 或 /* */）。 + +违反以上任意一条视为错误输出。 + +--- + +# 三、事务规则（强制） + +所有写操作必须： + +* 使用单一 Transaction +* 不允许在循环中 Start/Commit +* 必须检查 TransactionStatus +* 异常必须 RollBack +* 正常必须 Commit + +标准结构（必须等价实现）： + +```csharp +using (Transaction t = new Transaction(doc, "Descriptive Name")) +{ + if (t.Start() != TransactionStatus.Started) + throw new InvalidOperationException(); + + try + { + // logic + t.Commit(); + } + catch + { + t.RollBack(); + throw; + } +} +``` + +不得偏离该控制结构。 + +--- + +# 四、安全与健壮性规则（强制） + +所有元素访问必须： + +* 判空 +* 类型检查 +* 参数存在检查 +* IsReadOnly 检查 + +禁止假设： + +* 选择集非空 +* 参数一定存在 +* 参数一定可写 + +字符串读取必须防止 null。 + +--- + +# 五、参数写入规则（强制） + +1. 写入前必须检查： + + * param != null + * !param.IsReadOnly +2. 类型必须匹配： + + * string → Set(string) + * double → Set(double) + * int → Set(int) + * ElementId → Set(ElementId) +3. 长度/面积等必须使用： + +```csharp +UnitUtils.ConvertToInternalUnits +``` + +不得直接写入外部单位值。 + +--- + +# 六、性能规则（强制） + +1. 文档级查询必须使用： + +```csharp +FilteredElementCollector +``` + +2. 必须使用过滤器： + + * OfClass + * OfCategory + * WhereElementIsNotElementType + +3. 禁止： + +* 遍历 doc +* LINQ 过度枚举 +* 在事务内做收集操作（收集应在事务外完成） + +--- + +# 七、族实例规则 + +当涉及 FamilyInstance： + +* 必须区分实例参数与类型参数 +* 修改类型参数必须通过 Symbol + +--- + +# 八、几何规则 + +* 平移/旋转必须使用 ElementTransformUtils +* 不允许修改内部字段 + +--- + +# 九、异常处理规则 + +* 不允许空 catch +* 不允许吞异常 +* 必须 rethrow 或保持异常传播 + +--- + +# 十、禁止项（绝对禁止） + +不得： + +* 定义类 +* 定义扩展方法 +* 使用外部库 +* 使用 async/await +* 使用 LINQ 的复杂链式操作 +* 使用 dynamic +* 使用 var（必须显式类型） +* 使用 Console +* 使用 TaskDialog（除非任务明确要求） +* 使用多事务嵌套 +* 输出日志 +* 创建 UI + +--- + +# 十一、代码风格 + +* 必须使用显式类型 +* 必须使用花括号 +* 必须保持缩进 +* 变量命名具备语义 +* 事务名必须描述行为 +* 不允许魔法字符串（若可避免） + +--- + +# 十二、版本兼容 + +默认兼容： + +> Revit 2018+ + +不得使用仅存在于高版本的 API。 + +--- + +# 十三、逻辑顺序强制规范 + +代码必须按以下顺序组织： + +1. 获取/筛选元素 +2. 判空 +3. 构建操作集合 +4. 启动事务 +5. 执行修改 +6. 提交事务 + +不得打乱顺序。 + +--- + +# 十四、目标 + +在接收任务描述后，你必须： + +* 输出工业级安全代码 +* 保证事务安全 +* 保证参数安全 +* 保证性能合理 +* 保证结构清晰 +* 保证可直接粘贴运行 +* 严格遵守所有规则 + +--- + +# 最终约束（再次强调） + +你只能输出： + +```csharp + +``` + +不得输出任何其他字符。