feat: enhance LLM functionality with nginx configuration context and update ESLint auto-imports

Author: 0xJacky

.claude/settings.local.json

@@ -14,7 +14,8 @@
       "mcp__context7__get-library-docs",
       "Bash(find:*)",
       "Bash(sed:*)",
-      "Bash(cp:*)"
+      "Bash(cp:*)",
+      "mcp__eslint__lint-files"
     ],
     "deny": []
   }

api/llm/llm.go

@@ -17,35 +17,51 @@ import (
 	"github.com/uozi-tech/cosy/logger"
 )
 
-const LLMInitPrompt = `You are a assistant who can help users write and optimise the configurations of Nginx,
-the first user message contains the content of the configuration file which is currently opened by the user and
-the current language code(CLC). You suppose to use the language corresponding to the CLC to give the first reply.
-Later the language environment depends on the user message.
-The first reply should involve the key information of the file and ask user what can you help them.`
 
 func MakeChatCompletionRequest(c *gin.Context) {
 	var json struct {
-		Filepath string                         `json:"filepath"`
-		Messages []openai.ChatCompletionMessage `json:"messages"`
+		Type        string                         `json:"type"`
+		Messages    []openai.ChatCompletionMessage `json:"messages"`
+		Language    string                         `json:"language,omitempty"`
+		NginxConfig string                         `json:"nginx_config,omitempty"` // Separate field for nginx configuration content
 	}
 
 	if !cosy.BindAndValid(c, &json) {
 		return
 	}
 
+	// Choose appropriate system prompt based on the type
+	var systemPrompt string
+	if json.Type == "terminal" {
+		systemPrompt = llm.TerminalAssistantPrompt
+	} else {
+		systemPrompt = llm.NginxConfigPrompt
+	}
+
+	// Append language instruction if language is provided
+	if json.Language != "" {
+		systemPrompt += fmt.Sprintf("\n\nIMPORTANT: Please respond in the language corresponding to this language code: %s", json.Language)
+	}
+
 	messages := []openai.ChatCompletionMessage{
 		{
 			Role:    openai.ChatMessageRoleSystem,
-			Content: LLMInitPrompt,
+			Content: systemPrompt,
 		},
 	}
 
-	messages = append(messages, json.Messages...)
-
-	if json.Filepath != "" {
-		messages = llm.ChatCompletionWithContext(json.Filepath, messages)
+	// Add nginx configuration context if provided
+	if json.Type != "terminal" && json.NginxConfig != "" {
+		// Add nginx configuration as context to the first user message
+		if len(json.Messages) > 0 && json.Messages[0].Role == openai.ChatMessageRoleUser {
+			// Prepend the nginx configuration to the first user message
+			contextualContent := fmt.Sprintf("Nginx Configuration:\n```nginx\n%s\n```\n\n%s", json.NginxConfig, json.Messages[0].Content)
+			json.Messages[0].Content = contextualContent
+		}
 	}
 
+	messages = append(messages, json.Messages...)
+
 	// SSE server
 	api.SetSSEHeaders(c)
 

api/llm/session.go

@@ -15,14 +15,26 @@ import (
 	"github.com/uozi-tech/cosy/logger"
 )
 
+const TerminalAssistantPath = "__terminal_assistant__"
+
 // GetLLMSessions returns LLM sessions with optional filtering
 func GetLLMSessions(c *gin.Context) {
 	g := query.LLMSession
 	query := g.Order(g.UpdatedAt.Desc())
 
-	// Filter by path if provided
-	if path := c.Query("path"); path != "" {
-		if !helper.IsUnderDirectory(path, nginx.GetConfPath()) {
+	// Filter by type if provided
+	if assistantType := c.Query("type"); assistantType != "" {
+		if assistantType == "terminal" {
+			// For terminal type, filter by terminal assistant path
+			query = query.Where(g.Path.Eq(TerminalAssistantPath))
+		} else if assistantType == "nginx" {
+			// For nginx type, exclude terminal assistant path
+			query = query.Where(g.Path.Neq(TerminalAssistantPath))
+		}
+	} else if path := c.Query("path"); path != "" {
+		// Filter by path if provided (legacy support)
+		// Skip path validation for terminal assistant
+		if path != TerminalAssistantPath && !helper.IsUnderDirectory(path, nginx.GetConfPath()) {
 			c.JSON(http.StatusForbidden, gin.H{
 				"message": "path is not under the nginx conf path",
 			})
@@ -59,23 +71,31 @@ func CreateLLMSession(c *gin.Context) {
 	var json struct {
 		Title string `json:"title" binding:"required"`
 		Path  string `json:"path"`
+		Type  string `json:"type"`
 	}
 
 	if !cosy.BindAndValid(c, &json) {
 		return
 	}
 
-	// Validate path if provided
-	if json.Path != "" && !helper.IsUnderDirectory(json.Path, nginx.GetConfPath()) {
-		c.JSON(http.StatusForbidden, gin.H{
-			"message": "path is not under the nginx conf path",
-		})
-		return
+	// Determine path based on type
+	var sessionPath string
+	if json.Type == "terminal" {
+		sessionPath = TerminalAssistantPath
+	} else {
+		sessionPath = json.Path
+		// Validate path for non-terminal types
+		if sessionPath != "" && !helper.IsUnderDirectory(sessionPath, nginx.GetConfPath()) {
+			c.JSON(http.StatusForbidden, gin.H{
+				"message": "path is not under the nginx conf path",
+			})
+			return
+		}
 	}
 
 	session := &model.LLMSession{
 		Title:        json.Title,
-		Path:         json.Path,
+		Path:         sessionPath,
 		Messages:     []openai.ChatCompletionMessage{},
 		MessageCount: 0,
 		IsActive:     true,
@@ -197,7 +217,8 @@ func DuplicateLLMSession(c *gin.Context) {
 func GetLLMSessionByPath(c *gin.Context) {
 	path := c.Query("path")
 
-	if !helper.IsUnderDirectory(path, nginx.GetConfPath()) {
+	// Skip path validation for terminal assistant
+	if path != TerminalAssistantPath && !helper.IsUnderDirectory(path, nginx.GetConfPath()) {
 		c.JSON(http.StatusForbidden, gin.H{
 			"message": "path is not under the nginx conf path",
 		})
@@ -250,7 +271,8 @@ func CreateOrUpdateLLMSessionByPath(c *gin.Context) {
 		return
 	}
 
-	if !helper.IsUnderDirectory(json.FileName, nginx.GetConfPath()) {
+	// Skip path validation for terminal assistant
+	if json.FileName != TerminalAssistantPath && !helper.IsUnderDirectory(json.FileName, nginx.GetConfPath()) {
 		c.JSON(http.StatusForbidden, gin.H{
 			"message": "path is not under the nginx conf path",
 		})

app/.eslint-auto-import.mjs

@@ -4,6 +4,7 @@ export default {
     "$ngettext": true,
     "$npgettext": true,
     "$pgettext": true,
+    "App": true,
     "Component": true,
     "ComponentPublicInstance": true,
     "ComputedRef": true,

app/auto-imports.d.ts

@@ -10,6 +10,7 @@ declare global {
   const $ngettext: typeof import('@/gettext')['$ngettext']
   const $npgettext: typeof import('@/gettext')['$npgettext']
   const $pgettext: typeof import('@/gettext')['$pgettext']
+  const App: typeof import('ant-design-vue')['App']
   const EffectScope: typeof import('vue')['EffectScope']
   const T: typeof import('@/language')['T']
   const acceptHMRUpdate: typeof import('pinia')['acceptHMRUpdate']
@@ -38,7 +39,10 @@ declare global {
   const mapStores: typeof import('pinia')['mapStores']
   const mapWritableState: typeof import('pinia')['mapWritableState']
   const markRaw: typeof import('vue')['markRaw']
+  const message: typeof import('@/useAntApp')['message']
+  const modal: typeof import('@/useAntApp')['modal']
   const nextTick: typeof import('vue')['nextTick']
+  const notification: typeof import('@/useAntApp')['notification']
   const onActivated: typeof import('vue')['onActivated']
   const onBeforeMount: typeof import('vue')['onBeforeMount']
   const onBeforeRouteLeave: typeof import('vue-router')['onBeforeRouteLeave']
@@ -72,6 +76,9 @@ declare global {
   const toValue: typeof import('vue')['toValue']
   const triggerRef: typeof import('vue')['triggerRef']
   const unref: typeof import('vue')['unref']
+  const useAntAppStore: typeof import('@/pinia')['useAntAppStore']
+  const useApp: typeof import('ant-design-vue')['App.useApp']
+  const useAppUtils: typeof import('@/useAntApp')['useAppUtils']
   const useAttrs: typeof import('vue')['useAttrs']
   const useCssModule: typeof import('vue')['useCssModule']
   const useCssVars: typeof import('vue')['useCssVars']
@@ -103,6 +110,7 @@ declare module 'vue' {
     readonly $ngettext: UnwrapRef<typeof import('@/gettext')['$ngettext']>
     readonly $npgettext: UnwrapRef<typeof import('@/gettext')['$npgettext']>
     readonly $pgettext: UnwrapRef<typeof import('@/gettext')['$pgettext']>
+    readonly App: UnwrapRef<typeof import('ant-design-vue')['App']>
     readonly EffectScope: UnwrapRef<typeof import('vue')['EffectScope']>
     readonly T: UnwrapRef<typeof import('@/language')['T']>
     readonly acceptHMRUpdate: UnwrapRef<typeof import('pinia')['acceptHMRUpdate']>

app/components.d.ts

@@ -9,6 +9,7 @@ export {}
 declare module 'vue' {
   export interface GlobalComponents {
     AAlert: typeof import('ant-design-vue/es')['Alert']
+    AApp: typeof import('ant-design-vue/es')['App']
     AAutoComplete: typeof import('ant-design-vue/es')['AutoComplete']
     AAvatar: typeof import('ant-design-vue/es')['Avatar']
     ABadge: typeof import('ant-design-vue/es')['Badge']
@@ -90,6 +91,7 @@ declare module 'vue' {
     LLMChatMessageInput: typeof import('./src/components/LLM/ChatMessageInput.vue')['default']
     LLMChatMessageList: typeof import('./src/components/LLM/ChatMessageList.vue')['default']
     LLMLLM: typeof import('./src/components/LLM/LLM.vue')['default']
+    LLMLLMIframe: typeof import('./src/components/LLM/LLMIframe.vue')['default']
     LLMLLMSessionSelector: typeof import('./src/components/LLM/LLMSessionSelector.vue')['default']
     LLMLLMSessionSidebar: typeof import('./src/components/LLM/LLMSessionSidebar.vue')['default']
     LLMLLMSessionTabs: typeof import('./src/components/LLM/LLMSessionTabs.vue')['default']
@@ -123,6 +125,7 @@ declare module 'vue' {
     SelfCheckSelfCheckHeaderBanner: typeof import('./src/components/SelfCheck/SelfCheckHeaderBanner.vue')['default']
     SensitiveStringSensitiveString: typeof import('./src/components/SensitiveString/SensitiveString.vue')['default']
     SetLanguageSetLanguage: typeof import('./src/components/SetLanguage/SetLanguage.vue')['default']
+    ShadowRootShadowRoot: typeof import('./src/components/ShadowRoot/ShadowRoot.vue')['default']
     SwitchAppearanceIconsVPIconMoon: typeof import('./src/components/SwitchAppearance/icons/VPIconMoon.vue')['default']
     SwitchAppearanceIconsVPIconSun: typeof import('./src/components/SwitchAppearance/icons/VPIconSun.vue')['default']
     SwitchAppearanceSwitchAppearance: typeof import('./src/components/SwitchAppearance/SwitchAppearance.vue')['default']

app/src/App.vue

@@ -59,7 +59,9 @@ loadTranslations(route)
     :locale="lang"
     :auto-insert-space-in-button="false"
   >
-    <RouterView />
+    <AApp>
+      <RouterView />
+    </AApp>
   </AConfigProvider>
 </template>
 

app/src/api/llm.ts

@@ -49,15 +49,24 @@ const llm = {
   },
 
   // Session APIs
-  get_sessions(path?: string) {
+  get_sessions(pathOrType?: string, isType?: boolean) {
+    const params: Record<string, string> = {}
+    if (pathOrType) {
+      if (isType) {
+        params.type = pathOrType
+      }
+      else {
+        params.path = pathOrType
+      }
+    }
     return http.get<LLMSessionResponse[]>('/llm_sessions', {
-      params: path ? { path } : undefined,
+      params: Object.keys(params).length > 0 ? params : undefined,
     })
   },
   get_session(sessionId: string) {
     return http.get<LLMSessionResponse>(`/llm_sessions/${sessionId}`)
   },
-  create_session(data: { title: string, path?: string }) {
+  create_session(data: { title: string, path?: string, type?: string }) {
     return http.post<LLMSessionResponse>('/llm_sessions', data)
   },
   update_session(sessionId: string, data: { title?: string, messages?: ChatComplicationMessage[] }) {