wip

Author: tastelikefeet

modelscope_agent/cli/code/coding.yaml

@@ -91,6 +91,7 @@ prompt:
 callbacks:
   - artifact_callback
   - evaluator_callback
+  - prompt_callback
 
 memory:
 

modelscope_agent/cli/code/prompt_callback.py

@@ -0,0 +1,78 @@
+import json
+from typing import List
+
+from omegaconf import DictConfig
+
+from modelscope_agent.callbacks import Callback, Runtime
+from modelscope_agent.llm.utils import Message
+from modelscope_agent.utils import get_logger
+
+logger = get_logger()
+
+
+class PromptCallback(Callback):
+
+    _prompt = """
+
+Here are some specific instructions for frontend design:
+
+* **Overall Style:** Consider magazine-style, publication-style, or other modern web design styles you deem appropriate. The goal is to create a page that is both informative and visually appealing, like a well-designed digital magazine or in-depth feature article.
+
+* **Hero Section (Optional but Strongly Recommended):** If you think it's appropriate, design an eye-catching Hero section. It can include a main headline, subtitle, an engaging introductory paragraph, and a high-quality background image or illustration.
+
+* **Typography:**
+  * Carefully select font combinations (serif and sans-serif) to enhance the reading experience.
+  * Use different font sizes, weights, colors, and styles to create a clear visual hierarchy.
+  * Consider using refined typographic details (such as drop caps, hanging punctuation) to enhance overall quality.
+  * Font Awesome has many icons - choose appropriate ones to add visual interest and playfulness.
+
+* **Color Scheme:**
+  * Choose a color palette that is both harmonious and visually impactful.
+  * Consider using high-contrast color combinations to highlight important elements.
+  * Explore gradients, shadows, and other effects to add visual depth.
+
+* **Layout:**
+  * Use a grid-based layout system to organize page elements.
+  * Make full use of negative space (whitespace) to create visual balance and breathing room.
+  * Consider using cards, dividers, icons, and other visual elements to separate and organize content.
+
+* **Tone:** Overall style should be refined, creating a sense of sophistication.
+
+* **Data Visualization:**
+  * Design one or more data visualization elements to showcase key concepts of Naval's thinking and their relationships.
+  * Consider using mind maps, concept relationship diagrams, timelines, or thematic clustering displays.
+  * Ensure visualization design is both beautiful and insightful, helping users intuitively understand the overall framework of Naval's thought system.
+  * Use Mermaid.js to implement interactive charts that allow users to explore connections between different concepts.
+
+**Technical Specifications:**
+
+*   Use HTML5、Font Awesome、Tailwind CSS and necessary JavaScript。
+    *   Font Awesome: [https://cdn.staticfile.org/font-awesome/6.4.0/css/all.min.css](https://cdn.staticfile.org/font-awesome/6.4.0/css/all.min.css)
+    *   Tailwind CSS: [https://cdn.staticfile.org/tailwindcss/2.2.19/tailwind.min.css](https://cdn.staticfile.org/tailwindcss/2.2.19/tailwind.min.css)
+    *   Font for non-Chinese: [https://fonts.googleapis.com/css2?family=Noto+Serif+SC:wght@400;500;600;700&family=Noto+Sans+SC:wght@300;400;500;700&display=swap](https://fonts.googleapis.com/css2?family=Noto+Serif+SC:wght@400;500;600;700&family=Noto+Sans+SC:wght@300;400;500;700&display=swap)
+    *   `font-family: Tahoma,Arial,Roboto,"Droid Sans","Helvetica Neue","Droid Sans Fallback","Heiti SC","Hiragino Sans GB",Simsun,sans-self;`
+    *   Mermaid: [https://cdn.jsdelivr.net/npm/mermaid@latest/dist/mermaid.min.js](https://cdn.jsdelivr.net/npm/mermaid@latest/dist/mermaid.min.js)
+
+* Implement a complete dark/light mode toggle functionality that follows system settings by default and allows users to manually switch.
+* Code structure should be clear and semantic, including appropriate comments.
+* Implement complete responsiveness that must display perfectly on all devices (mobile, tablet, desktop).
+"""
+
+    def __init__(self, config: DictConfig):
+        super().__init__(config)
+
+    async def on_tool_call(self, runtime: Runtime, messages: List[Message]):
+        if runtime.tag != 'Default workflow':
+            return
+
+        if messages[-1].tool_calls:
+            tool_call = messages[-1].tool_calls[0]
+            tool_args = tool_call['arguments']
+            if isinstance(tool_args, str):
+                tool_args = json.loads(tool_args)
+            tasks = tool_args['tasks']
+            for i, task in enumerate(tasks):
+                system = task['system']
+                system = system + self._prompt
+                task['system'] = system
+            tool_call['arguments'] = tool_args