soraacode 0.1.7 → 0.2.1

package/src/session/message.ts

@@ -1,148 +0,0 @@
-import { Schema } from "effect"
-import { SessionID } from "./schema"
-
-import { NonNegativeInt } from "@soraacode/core/schema"
-import { MessageError } from "./message-error"
-import { AuthError, OutputLengthError } from "./message-error"
-import { ProviderV2 } from "@soraacode/core/provider"
-import { ModelV2 } from "@soraacode/core/model"
-export { AuthError, OutputLengthError } from "./message-error"
-
-export const ToolCall = Schema.Struct({
-  state: Schema.Literal("call"),
-  step: Schema.optional(NonNegativeInt),
-  toolCallId: Schema.String,
-  toolName: Schema.String,
-  args: Schema.Unknown,
-}).annotate({ identifier: "ToolCall" })
-export type ToolCall = Schema.Schema.Type<typeof ToolCall>
-
-export const ToolPartialCall = Schema.Struct({
-  state: Schema.Literal("partial-call"),
-  step: Schema.optional(NonNegativeInt),
-  toolCallId: Schema.String,
-  toolName: Schema.String,
-  args: Schema.Unknown,
-}).annotate({ identifier: "ToolPartialCall" })
-export type ToolPartialCall = Schema.Schema.Type<typeof ToolPartialCall>
-
-export const ToolResult = Schema.Struct({
-  state: Schema.Literal("result"),
-  step: Schema.optional(NonNegativeInt),
-  toolCallId: Schema.String,
-  toolName: Schema.String,
-  args: Schema.Unknown,
-  result: Schema.String,
-}).annotate({ identifier: "ToolResult" })
-export type ToolResult = Schema.Schema.Type<typeof ToolResult>
-
-export const ToolInvocation = Schema.Union([ToolCall, ToolPartialCall, ToolResult]).annotate({
-  identifier: "ToolInvocation",
-  discriminator: "state",
-})
-export type ToolInvocation = Schema.Schema.Type<typeof ToolInvocation>
-
-export const TextPart = Schema.Struct({
-  type: Schema.Literal("text"),
-  text: Schema.String,
-}).annotate({ identifier: "TextPart" })
-export type TextPart = Schema.Schema.Type<typeof TextPart>
-
-export const ReasoningPart = Schema.Struct({
-  type: Schema.Literal("reasoning"),
-  text: Schema.String,
-  providerMetadata: Schema.optional(Schema.Record(Schema.String, Schema.Unknown)),
-}).annotate({ identifier: "ReasoningPart" })
-export type ReasoningPart = Schema.Schema.Type<typeof ReasoningPart>
-
-export const ToolInvocationPart = Schema.Struct({
-  type: Schema.Literal("tool-invocation"),
-  toolInvocation: ToolInvocation,
-}).annotate({ identifier: "ToolInvocationPart" })
-export type ToolInvocationPart = Schema.Schema.Type<typeof ToolInvocationPart>
-
-export const SourceUrlPart = Schema.Struct({
-  type: Schema.Literal("source-url"),
-  sourceId: Schema.String,
-  url: Schema.String,
-  title: Schema.optional(Schema.String),
-  providerMetadata: Schema.optional(Schema.Record(Schema.String, Schema.Unknown)),
-}).annotate({ identifier: "SourceUrlPart" })
-export type SourceUrlPart = Schema.Schema.Type<typeof SourceUrlPart>
-
-export const FilePart = Schema.Struct({
-  type: Schema.Literal("file"),
-  mediaType: Schema.String,
-  filename: Schema.optional(Schema.String),
-  url: Schema.String,
-}).annotate({ identifier: "FilePart" })
-export type FilePart = Schema.Schema.Type<typeof FilePart>
-
-export const StepStartPart = Schema.Struct({
-  type: Schema.Literal("step-start"),
-}).annotate({ identifier: "StepStartPart" })
-export type StepStartPart = Schema.Schema.Type<typeof StepStartPart>
-
-export const MessagePart = Schema.Union([
-  TextPart,
-  ReasoningPart,
-  ToolInvocationPart,
-  SourceUrlPart,
-  FilePart,
-  StepStartPart,
-]).annotate({ identifier: "MessagePart", discriminator: "type" })
-export type MessagePart = Schema.Schema.Type<typeof MessagePart>
-
-export const Info = Schema.Struct({
-  id: Schema.String,
-  role: Schema.Literals(["user", "assistant"]),
-  parts: Schema.Array(MessagePart),
-  metadata: Schema.Struct({
-    time: Schema.Struct({
-      created: NonNegativeInt,
-      completed: Schema.optional(NonNegativeInt),
-    }),
-    error: Schema.optional(MessageError.SharedSchema),
-    sessionID: SessionID,
-    tool: Schema.Record(
-      Schema.String,
-      Schema.StructWithRest(
-        Schema.Struct({
-          title: Schema.String,
-          snapshot: Schema.optional(Schema.String),
-          time: Schema.Struct({
-            start: NonNegativeInt,
-            end: NonNegativeInt,
-          }),
-        }),
-        [Schema.Record(Schema.String, Schema.Unknown)],
-      ),
-    ),
-    assistant: Schema.optional(
-      Schema.Struct({
-        system: Schema.Array(Schema.String),
-        modelID: ModelV2.ID,
-        providerID: ProviderV2.ID,
-        path: Schema.Struct({
-          cwd: Schema.String,
-          root: Schema.String,
-        }),
-        cost: Schema.Finite,
-        summary: Schema.optional(Schema.Boolean),
-        tokens: Schema.Struct({
-          input: Schema.Finite,
-          output: Schema.Finite,
-          reasoning: Schema.Finite,
-          cache: Schema.Struct({
-            read: Schema.Finite,
-            write: Schema.Finite,
-          }),
-        }),
-      }),
-    ),
-    snapshot: Schema.optional(Schema.String),
-  }).annotate({ identifier: "MessageMetadata" }),
-}).annotate({ identifier: "Message" })
-export type Info = Schema.Schema.Type<typeof Info>
-
-export * as Message from "./message"

package/src/session/overflow.ts

@@ -1,34 +0,0 @@
-import type { Config } from "@/config/config"
-import { ConfigV1 } from "@soraacode/core/v1/config/config"
-import { SessionV1 } from "@soraacode/core/v1/session"
-import type { Provider } from "@/provider/provider"
-import { ProviderTransform } from "@/provider/transform"
-import type { MessageV2 } from "./message-v2"
-
-const COMPACTION_BUFFER = 20_000
-
-export function usable(input: { cfg: ConfigV1.Info; model: Provider.Model; outputTokenMax?: number }) {
-  const context = input.model.limit.context
-  if (context === 0) return 0
-
-  const reserved =
-    input.cfg.compaction?.reserved ??
-    Math.min(COMPACTION_BUFFER, ProviderTransform.maxOutputTokens(input.model, input.outputTokenMax))
-  return input.model.limit.input
-    ? Math.max(0, input.model.limit.input - reserved)
-    : Math.max(0, context - ProviderTransform.maxOutputTokens(input.model, input.outputTokenMax))
-}
-
-export function isOverflow(input: {
-  cfg: ConfigV1.Info
-  tokens: SessionV1.Assistant["tokens"]
-  model: Provider.Model
-  outputTokenMax?: number
-}) {
-  if (input.cfg.compaction?.auto === false) return false
-  if (input.model.limit.context === 0) return false
-
-  const count =
-    input.tokens.total || input.tokens.input + input.tokens.output + input.tokens.cache.read + input.tokens.cache.write
-  return count >= usable(input)
-}