codebuff 1.0.281 → 1.0.282

package/dist/common/actions.d.ts

@@ -4,12 +4,12 @@ export declare const FileChangeSchema: z.ZodObject<{
     path: z.ZodString;
     content: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    type: "patch" | "file";
     path: string;
+    type: "patch" | "file";
     content: string;
 }, {
-    type: "patch" | "file";
     path: string;
+    type: "patch" | "file";
     content: string;
 }>;
 export type FileChange = z.infer<typeof FileChangeSchema>;
@@ -18,12 +18,12 @@ export declare const CHANGES: z.ZodArray<z.ZodObject<{
     path: z.ZodString;
     content: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    type: "patch" | "file";
     path: string;
+    type: "patch" | "file";
     content: string;
 }, {
-    type: "patch" | "file";
     path: string;
+    type: "patch" | "file";
     content: string;
 }>, "many">;
 export type FileChanges = z.infer<typeof CHANGES>;
@@ -199,16 +199,16 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 }>>;
             }, "strip", z.ZodTypeAny, {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
                     type: "ephemeral";
                 } | undefined;
             }, {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -290,8 +290,8 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -324,8 +324,8 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -390,8 +390,8 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -456,8 +456,8 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -504,7 +504,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     type: "prompt";
     promptId: string;
     fingerprintId: string;
-    costMode: "lite" | "normal" | "max" | "experimental";
+    costMode: "max" | "lite" | "normal" | "experimental";
     agentState: {
         agentContext: string;
         fileContext: {
@@ -543,8 +543,8 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -623,8 +623,8 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -659,7 +659,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     }[];
     prompt?: string | undefined;
     authToken?: string | undefined;
-    costMode?: "lite" | "normal" | "max" | "experimental" | undefined;
+    costMode?: "max" | "lite" | "normal" | "experimental" | undefined;
     model?: string | undefined;
     cwd?: string | undefined;
 }>, z.ZodObject<{
@@ -792,7 +792,6 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     }>;
 }, "strip", z.ZodTypeAny, {
     type: "init";
-    fingerprintId: string;
     fileContext: {
         currentWorkingDirectory: string;
         fileTree: import("./util/file").FileTreeNode[];
@@ -821,10 +820,10 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             content: string;
         }[][] | undefined;
     };
+    fingerprintId: string;
     authToken?: string | undefined;
 }, {
     type: "init";
-    fingerprintId: string;
     fileContext: {
         currentWorkingDirectory: string;
         fileTree: import("./util/file").FileTreeNode[];
@@ -853,6 +852,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             content: string;
         }[][] | undefined;
     };
+    fingerprintId: string;
     authToken?: string | undefined;
 }>, z.ZodObject<{
     type: z.ZodLiteral<"generate-commit-message">;
@@ -928,12 +928,12 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }>, "many">;
     changesAlreadyApplied: z.ZodArray<z.ZodObject<{
@@ -941,12 +941,12 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }>, "many">;
     addedFileVersions: z.ZodArray<z.ZodObject<{
@@ -971,13 +971,13 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
     userInputId: string;
     response: string;
     changes: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     changesAlreadyApplied: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     addedFileVersions: {
@@ -995,13 +995,13 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
     userInputId: string;
     response: string;
     changes: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     changesAlreadyApplied: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     addedFileVersions: {
@@ -1183,16 +1183,16 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 }>>;
             }, "strip", z.ZodTypeAny, {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
                     type: "ephemeral";
                 } | undefined;
             }, {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1274,8 +1274,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1308,8 +1308,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1374,8 +1374,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1440,8 +1440,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1475,12 +1475,12 @@ export declare const PromptResponseSchema: z.ZodObject<{
         id: z.ZodString;
     }, "strip", z.ZodTypeAny, {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }, {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }>, "many">;
     toolResults: z.ZodArray<z.ZodObject<{
         name: z.ZodString;
@@ -1536,8 +1536,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1572,8 +1572,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
     }[];
     toolCalls: {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }[];
 }, {
     type: "prompt-response";
@@ -1616,8 +1616,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -1652,8 +1652,8 @@ export declare const PromptResponseSchema: z.ZodObject<{
     }[];
     toolCalls: {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }[];
 }>;
 export type PromptResponse = z.infer<typeof PromptResponseSchema>;
@@ -1678,12 +1678,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }>, "many">;
     changesAlreadyApplied: z.ZodArray<z.ZodObject<{
@@ -1691,12 +1691,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }>, "many">;
     addedFileVersions: z.ZodArray<z.ZodObject<{
@@ -1721,13 +1721,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     changesAlreadyApplied: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     addedFileVersions: {
@@ -1745,13 +1745,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     changesAlreadyApplied: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     addedFileVersions: {
@@ -1918,16 +1918,16 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 }>>;
             }, "strip", z.ZodTypeAny, {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
                     type: "ephemeral";
                 } | undefined;
             }, {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2009,8 +2009,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2043,8 +2043,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2109,8 +2109,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2175,8 +2175,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2210,12 +2210,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         id: z.ZodString;
     }, "strip", z.ZodTypeAny, {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }, {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }>, "many">;
     toolResults: z.ZodArray<z.ZodObject<{
         name: z.ZodString;
@@ -2271,8 +2271,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2307,8 +2307,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     }[];
     toolCalls: {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }[];
 }, {
     type: "prompt-response";
@@ -2351,8 +2351,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                     type: "ephemeral";
                 } | undefined;
             } | {
-                type: "tool_use";
                 name: string;
+                type: "tool_use";
                 id: string;
                 input: Record<string, any>;
                 cache_control?: {
@@ -2387,8 +2387,8 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     }[];
     toolCalls: {
         name: string;
-        id: string;
         parameters: Record<string, string>;
+        id: string;
     }[];
 }>, z.ZodObject<{
     type: z.ZodLiteral<"read-files">;
@@ -2424,12 +2424,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }>, "many">;
     changesAlreadyApplied: z.ZodArray<z.ZodObject<{
@@ -2437,12 +2437,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }, {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }>, "many">;
     addedFileVersions: z.ZodArray<z.ZodObject<{
@@ -2466,13 +2466,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     changesAlreadyApplied: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     addedFileVersions: {
@@ -2490,13 +2490,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     changesAlreadyApplied: {
-        type: "patch" | "file";
         path: string;
+        type: "patch" | "file";
         content: string;
     }[];
     addedFileVersions: {
@@ -2590,13 +2590,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     error: z.ZodOptional<z.ZodString>;
     remainingBalance: z.ZodOptional<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
-    type: "action-error";
     message: string;
+    type: "action-error";
     remainingBalance?: number | undefined;
     error?: string | undefined;
 }, {
-    type: "action-error";
     message: string;
+    type: "action-error";
     remainingBalance?: number | undefined;
     error?: string | undefined;
 }>, z.ZodObject<{

package/dist/common/bigquery/client.d.ts

@@ -0,0 +1,15 @@
+import { GetRelevantFilesTrace, Relabel, Trace } from './schema';
+export declare function setupBigQuery(dataset?: string): Promise<void>;
+export declare function insertTrace(trace: Trace, dataset?: string): Promise<boolean>;
+export declare function insertRelabel(relabel: Relabel, dataset?: string): Promise<boolean>;
+export declare function getRecentTraces(limit?: number, dataset?: string): Promise<Trace[]>;
+export declare function getRecentRelabels(limit?: number, dataset?: string): Promise<Relabel[]>;
+export declare function getTracesWithoutRelabels(model: string, limit?: number, userId?: string | undefined, dataset?: string): Promise<GetRelevantFilesTrace[]>;
+export declare function getTracesWithRelabels(model: string, limit?: number, dataset?: string): Promise<{
+    trace: GetRelevantFilesTrace;
+    relabel: Relabel;
+}[]>;
+export declare function getTracesAndRelabelsForUser(userId: string, limit?: number, dataset?: string): Promise<{
+    trace: GetRelevantFilesTrace;
+    relabels: Relabel[];
+}[]>;