✨: Enhance type imports across multiple files

Added new type imports for better typing and code reliability.

Author: pelikhan

packages/core/src/agent.ts

@@ -9,7 +9,11 @@ import { HTMLEscape } from "./htmlescape.js";
 import { prettifyMarkdown } from "./markdown.js";
 import { TraceOptions } from "./trace.js";
 import { ellipse } from "./util.js";
-import type { ChatGenerationContext, WorkspaceFileCache } from "./types.js";
+import type {
+  ChatGenerationContext,
+  ChatTurnGenerationContext,
+  WorkspaceFileCache,
+} from "./types.js";
 
 import debug from "debug";
 const dbg = debug("agent:memory");
@@ -50,7 +54,6 @@ export async function agentQueryMemory(
   cache: AgentMemoryCache,
   ctx: ChatGenerationContext,
   query: string,
-  _options: Required<TraceOptions>,
 ) {
   if (!query) return undefined;
 

packages/core/src/annotations.ts

@@ -10,7 +10,7 @@
 import { deleteUndefinedValues } from "./cleaners.js";
 import { EMOJI_FAIL, EMOJI_WARNING } from "./constants.js";
 import { unfence } from "./unwrappers.js";
-import type { Diagnostic } from "./types.js";
+import type { Diagnostic, DiagnosticSeverity } from "./types.js";
 
 // Regular expression for matching GitHub Actions annotations.
 // Example: ::error file=foo.js,line=10,endLine=11::Something went wrong.
@@ -39,7 +39,7 @@ const GITHUB_MARKDOWN_WARNINGS_RX =
 // Example: src/connection.ts(71,5): error TS1128: Declaration or statement expected.
 // src/connection.ts(71,5): error TS1128: Declaration or statement expected.
 const TYPESCRIPT_PARENTHESES_ANNOTATIONS_RX =
-  /^(?<file>[^\(\n]+)\((?<line>\d+),(?<col>\d+)\):\s+(?<severity>error|warning)\s+(?<code>TS\d+):\s+(?<message>.+)$/gim;
+  /^(?<file>[^(\n]+)\((?<line>\d+),(?<col>\d+)\):\s+(?<severity>error|warning)\s+(?<code>TS\d+):\s+(?<message>.+)$/gim;
 const ANNOTATIONS_RX = [
   TYPESCRIPT_PARENTHESES_ANNOTATIONS_RX,
   TYPESCRIPT_ANNOTATIONS_RX,
@@ -187,6 +187,7 @@ export function convertGithubMarkdownAnnotationsToItems(text: string) {
  * @returns A formatted string representing the Diagnostic as a list item.
  */
 export function convertAnnotationToItem(d: Diagnostic) {
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   const { severity, message, filename, code, range } = d;
   const line = range?.[0]?.[0];
   return `- ${SEV_EMOJI_MAP[severity?.toLowerCase()] ?? "info"} ${message}${filename ? ` (\`${filename}${line ? `#L${line}` : ""}\`)` : ""}`;

packages/core/src/anthropic.ts

@@ -40,6 +40,7 @@ import { LanguageModelConfiguration } from "./server/messages.js";
 import { deleteUndefinedValues } from "./cleaners.js";
 import debug from "debug";
 import { providerFeatures } from "./features.js";
+import { LanguageModelInfo } from "./types.js";
 const dbg = debug("genaiscript:anthropic");
 const dbgMessages = debug("genaiscript:anthropic:msg");
 

packages/core/src/ast.ts

@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 
+/// <reference types="./types/prompt_template.d.ts" />
 // Import necessary regular expressions for file type detection and host utilities
 import { GENAI_ANYJS_REGEX, GENAI_ANYTS_REGEX } from "./constants.js";
 import { Project } from "./server/messages.js";

packages/core/src/azureaisearch.ts

@@ -14,6 +14,15 @@ import { LanguageModelConfiguration } from "./server/messages.js";
 import { chunk } from "./encoders.js";
 import { genaiscriptDebug } from "./debug.js";
 import { SearchClient, SearchIndexClient, AzureKeyCredential } from "@azure/search-documents";
+import type {
+  ElementOrArray,
+  TextChunk,
+  VectorIndexOptions,
+  VectorSearchOptions,
+  WorkspaceFile,
+  WorkspaceFileIndex,
+  WorkspaceFileWithScore,
+} from "./types.js";
 
 const dbg = genaiscriptDebug("azureaisearch");
 

packages/core/src/azuredevops.ts

@@ -110,6 +110,7 @@ export async function azureDevOpsUpdatePullRequestDescription(
   // query pull request
   const pr = await findPullRequest(info);
   if (!pr) return;
+  // eslint-disable-next-line prefer-const
   let { pullRequestId, description } = pr;
 
   text = prettifyMarkdown(text);

packages/core/src/azureopenai.ts

@@ -34,7 +34,7 @@ const azureManagementListModels: ListModelsFunction = async (cfg, options) => {
     const { base } = cfg;
     const subscriptionId = process.env.AZURE_OPENAI_SUBSCRIPTION_ID;
     let resourceGroupName = process.env.AZURE_OPENAI_RESOURCE_GROUP;
-    const accountName = /^https:\/\/([^\.]+)\./.exec(base)[1];
+    const accountName = /^https:\/\/([^.]+)\./.exec(base)[1];
 
     if (!subscriptionId || !accountName) {
       dbg("subscriptionId or accountName is missing, returning an empty model list");
@@ -78,7 +78,7 @@ const azureManagementListModels: ListModelsFunction = async (cfg, options) => {
         `https://management.azure.com/subscriptions/${subscriptionId}/resources?api-version=2021-04-01`,
       );
       const resource = resources.value.find((r) => r.name === accountName);
-      resourceGroupName = /\/resourceGroups\/([^\/]+)\/providers\//.exec(resource?.id)[1];
+      resourceGroupName = /\/resourceGroups\/([^/]+)\/providers\//.exec(resource?.id)[1];
       if (!resourceGroupName) {
         dbg("unable to extract resource group name from resource id");
         throw new Error("Resource group not found");

packages/core/src/azuretoken.ts

@@ -26,6 +26,7 @@ import {
   ManagedIdentityCredential,
   WorkloadIdentityCredential,
 } from "@azure/identity";
+import type { SerializedError } from "./types.js";
 
 /**
  * This module provides functions to handle Azure authentication tokens,

packages/core/src/chat.ts

@@ -97,6 +97,18 @@ import type {
   TranscriptionResult,
   VectorIndexOptions,
   WorkspaceFileIndex,
+  ChatMessage,
+  DataFrame,
+  Edits,
+  ElementOrArray,
+  Logprob,
+  RunPromptUsage,
+  ShellOutput,
+  TokenEncoder,
+  ToolCallContent,
+  ToolCallContext,
+  ToolCallOutput,
+  WorkspaceFile,
 } from "./types.js";
 
 const dbg = genaiscriptDebug("chat");

packages/core/src/mcpclient.ts

@@ -19,7 +19,7 @@ import { dotGenaiscriptPath } from "./workdir.js";
 import { YAMLStringify } from "./yaml.js";
 import { resolvePromptInjectionDetector } from "./contentsafety.js";
 import { genaiscriptDebug } from "./debug.js";
-import type { JSONSchemaObject, McpClient, McpServerConfig, ToolCallback } from "./types.js";
+import type { McpClient, McpServerConfig, ToolCallback } from "./types.js";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";