codebuff 1.0.273 → 1.0.275

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, {
-    path: string;
     type: "file" | "patch";
+    path: string;
     content: string;
 }, {
-    path: string;
     type: "file" | "patch";
+    path: string;
     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, {
-    path: string;
     type: "file" | "patch";
+    path: string;
     content: string;
 }, {
-    path: string;
     type: "file" | "patch";
+    path: string;
     content: string;
 }>, "many">;
 export type FileChanges = z.infer<typeof CHANGES>;
@@ -54,6 +54,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             currentWorkingDirectory: z.ZodString;
             fileTree: z.ZodArray<z.ZodType<import("./util/file").FileTreeNode, z.ZodTypeDef, import("./util/file").FileTreeNode>, "many">;
             fileTokenScores: z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodNumber>>;
+            tokenCallers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodArray<z.ZodString, "many">>>>;
             knowledgeFiles: z.ZodRecord<z.ZodString, z.ZodString>;
             userKnowledgeFiles: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
             gitChanges: z.ZodObject<{
@@ -127,6 +128,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -153,6 +155,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -372,6 +375,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -437,6 +441,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -523,6 +528,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -602,6 +608,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -675,6 +682,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         currentWorkingDirectory: z.ZodString;
         fileTree: z.ZodArray<z.ZodType<import("./util/file").FileTreeNode, z.ZodTypeDef, import("./util/file").FileTreeNode>, "many">;
         fileTokenScores: z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodNumber>>;
+        tokenCallers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodArray<z.ZodString, "many">>>>;
         knowledgeFiles: z.ZodRecord<z.ZodString, z.ZodString>;
         userKnowledgeFiles: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
         gitChanges: z.ZodObject<{
@@ -748,6 +756,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             homedir: string;
             cpus: number;
         };
+        tokenCallers?: Record<string, Record<string, string[]>> | undefined;
         userKnowledgeFiles?: Record<string, string> | undefined;
         fileVersions?: {
             path: string;
@@ -774,6 +783,7 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             homedir: string;
             cpus: number;
         };
+        tokenCallers?: Record<string, Record<string, string[]>> | undefined;
         userKnowledgeFiles?: Record<string, string> | undefined;
         fileVersions?: {
             path: string;
@@ -782,6 +792,7 @@ 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[];
@@ -803,16 +814,17 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             homedir: string;
             cpus: number;
         };
+        tokenCallers?: Record<string, Record<string, string[]>> | undefined;
         userKnowledgeFiles?: Record<string, string> | undefined;
         fileVersions?: {
             path: string;
             content: string;
         }[][] | undefined;
     };
-    fingerprintId: string;
     authToken?: string | undefined;
 }, {
     type: "init";
+    fingerprintId: string;
     fileContext: {
         currentWorkingDirectory: string;
         fileTree: import("./util/file").FileTreeNode[];
@@ -834,13 +846,13 @@ export declare const CLIENT_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             homedir: string;
             cpus: number;
         };
+        tokenCallers?: Record<string, Record<string, string[]>> | undefined;
         userKnowledgeFiles?: Record<string, string> | undefined;
         fileVersions?: {
             path: string;
             content: string;
         }[][] | undefined;
     };
-    fingerprintId: string;
     authToken?: string | undefined;
 }>, z.ZodObject<{
     type: z.ZodLiteral<"generate-commit-message">;
@@ -916,12 +928,12 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }>, "many">;
     changesAlreadyApplied: z.ZodArray<z.ZodObject<{
@@ -929,12 +941,12 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }>, "many">;
     addedFileVersions: z.ZodArray<z.ZodObject<{
@@ -959,13 +971,13 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
     userInputId: string;
     response: string;
     changes: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     changesAlreadyApplied: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     addedFileVersions: {
@@ -983,13 +995,13 @@ export declare const ResponseCompleteSchema: z.ZodObject<z.objectUtil.extendShap
     userInputId: string;
     response: string;
     changes: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     changesAlreadyApplied: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     addedFileVersions: {
@@ -1026,6 +1038,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
             currentWorkingDirectory: z.ZodString;
             fileTree: z.ZodArray<z.ZodType<import("./util/file").FileTreeNode, z.ZodTypeDef, import("./util/file").FileTreeNode>, "many">;
             fileTokenScores: z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodNumber>>;
+            tokenCallers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodArray<z.ZodString, "many">>>>;
             knowledgeFiles: z.ZodRecord<z.ZodString, z.ZodString>;
             userKnowledgeFiles: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
             gitChanges: z.ZodObject<{
@@ -1099,6 +1112,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1125,6 +1139,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1344,6 +1359,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1409,6 +1425,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1504,6 +1521,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1583,6 +1601,7 @@ export declare const PromptResponseSchema: z.ZodObject<{
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1659,12 +1678,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }>, "many">;
     changesAlreadyApplied: z.ZodArray<z.ZodObject<{
@@ -1672,12 +1691,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }>, "many">;
     addedFileVersions: z.ZodArray<z.ZodObject<{
@@ -1702,13 +1721,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     changesAlreadyApplied: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     addedFileVersions: {
@@ -1726,13 +1745,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     changesAlreadyApplied: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     addedFileVersions: {
@@ -1754,6 +1773,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
             currentWorkingDirectory: z.ZodString;
             fileTree: z.ZodArray<z.ZodType<import("./util/file").FileTreeNode, z.ZodTypeDef, import("./util/file").FileTreeNode>, "many">;
             fileTokenScores: z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodNumber>>;
+            tokenCallers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodArray<z.ZodString, "many">>>>;
             knowledgeFiles: z.ZodRecord<z.ZodString, z.ZodString>;
             userKnowledgeFiles: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
             gitChanges: z.ZodObject<{
@@ -1827,6 +1847,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -1853,6 +1874,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -2072,6 +2094,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -2137,6 +2160,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -2232,6 +2256,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -2311,6 +2336,7 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
                 homedir: string;
                 cpus: number;
             };
+            tokenCallers?: Record<string, Record<string, string[]>> | undefined;
             userKnowledgeFiles?: Record<string, string> | undefined;
             fileVersions?: {
                 path: string;
@@ -2398,12 +2424,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }>, "many">;
     changesAlreadyApplied: z.ZodArray<z.ZodObject<{
@@ -2411,12 +2437,12 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
         path: z.ZodString;
         content: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }, {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }>, "many">;
     addedFileVersions: z.ZodArray<z.ZodObject<{
@@ -2440,13 +2466,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     changesAlreadyApplied: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     addedFileVersions: {
@@ -2464,13 +2490,13 @@ export declare const SERVER_ACTION_SCHEMA: z.ZodDiscriminatedUnion<"type", [z.Zo
     userInputId: string;
     response: string;
     changes: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     changesAlreadyApplied: {
-        path: string;
         type: "file" | "patch";
+        path: string;
         content: string;
     }[];
     addedFileVersions: {
@@ -2564,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, {
-    message: string;
     type: "action-error";
+    message: string;
     remainingBalance?: number | undefined;
     error?: string | undefined;
 }, {
-    message: string;
     type: "action-error";
+    message: string;
     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[];
+}[]>;