@relayfx/sdk 0.3.3 → 0.3.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/ai.js +1 -1
- package/dist/http-server.js +1 -1
- package/dist/{index-49fm2rdv.js → index-5dqkw53p.js} +1 -1
- package/dist/{index-25gwk9tj.js → index-8fpd6kvj.js} +46 -31
- package/dist/{index-8pa5e2yn.js → index-cq8t8ka1.js} +217 -185
- package/dist/{index-cphmds30.js → index-cyfx5y3r.js} +3 -0
- package/dist/index.js +10 -9
- package/dist/migrations/20260718140000_execution_scoped_tool_calls/migration.sql +35 -0
- package/dist/migrations/mysql/0011_execution_scoped_tool_calls.sql +29 -0
- package/dist/migrations/pg/20260718140000_execution_scoped_tool_calls/migration.sql +35 -0
- package/dist/migrations/sqlite/0011_execution_scoped_tool_calls.sql +53 -0
- package/dist/mysql.js +4 -4
- package/dist/postgres.js +21 -10
- package/dist/sqlite.js +62 -5
- package/dist/types/relay/operation.d.ts +9 -0
- package/dist/types/relay/runtime.d.ts +2 -0
- package/dist/types/relay/sqlite-migrations.d.ts +6 -1
- package/dist/types/runtime/agent/agent-loop-service.d.ts +5 -1
- package/dist/types/runtime/runner/runner-runtime-service.d.ts +4 -0
- package/dist/types/store-sql/schema/relay-schema.d.ts +30 -0
- package/dist/types/store-sql/tool/tool-call-repository.d.ts +11 -6
- package/package.json +3 -3
|
@@ -16,7 +16,7 @@ import {
|
|
|
16
16
|
exports_tool_executor,
|
|
17
17
|
exports_tool_output,
|
|
18
18
|
exports_turn_policy
|
|
19
|
-
} from "./index-
|
|
19
|
+
} from "./index-8fpd6kvj.js";
|
|
20
20
|
import {
|
|
21
21
|
ClientError,
|
|
22
22
|
CommandReplyInvalid,
|
|
@@ -51,7 +51,7 @@ import {
|
|
|
51
51
|
followExecutionFrom,
|
|
52
52
|
inputSchemaRef,
|
|
53
53
|
outputSchemaRef
|
|
54
|
-
} from "./index-
|
|
54
|
+
} from "./index-cyfx5y3r.js";
|
|
55
55
|
import {
|
|
56
56
|
__export
|
|
57
57
|
} from "./index-nb39b5ae.js";
|
|
@@ -7679,9 +7679,10 @@ var MaxAttemptRow = Schema35.Struct({
|
|
|
7679
7679
|
});
|
|
7680
7680
|
var ExpiredCallRow = Schema35.Struct({
|
|
7681
7681
|
id: Schema35.String,
|
|
7682
|
+
execution_id: Schema35.String,
|
|
7682
7683
|
active_attempt_id: Schema35.NullishOr(Schema35.String)
|
|
7683
7684
|
});
|
|
7684
|
-
var IdRow = Schema35.Struct({ id: Schema35.String });
|
|
7685
|
+
var IdRow = Schema35.Struct({ id: Schema35.String, execution_id: Schema35.String });
|
|
7685
7686
|
var metadata8 = (input) => input ?? {};
|
|
7686
7687
|
var normalizeCall = (call) => ({ ...call, metadata: metadata8(call.metadata) });
|
|
7687
7688
|
var sameCallIdentity = (left, right) => exports_shared_schema.canonicalEquals(normalizeCall(left), normalizeCall(right));
|
|
@@ -7776,15 +7777,15 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7776
7777
|
"completed_at"
|
|
7777
7778
|
].join(", "));
|
|
7778
7779
|
const resultColumns = sql.literal(["tool_call_id", "output_json", "error", "metadata_json", "created_at"].join(", "));
|
|
7779
|
-
const selectCallByKey = (tenantId, id) => sql`
|
|
7780
|
+
const selectCallByKey = (tenantId, executionId, id) => sql`
|
|
7780
7781
|
SELECT ${callColumns}
|
|
7781
7782
|
FROM relay_tool_calls
|
|
7782
|
-
WHERE tenant_id = ${tenantId} AND id = ${id}
|
|
7783
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND id = ${id}
|
|
7783
7784
|
`;
|
|
7784
|
-
const selectResultByKey = (tenantId, callId) => sql`
|
|
7785
|
+
const selectResultByKey = (tenantId, executionId, callId) => sql`
|
|
7785
7786
|
SELECT ${resultColumns}
|
|
7786
7787
|
FROM relay_tool_results
|
|
7787
|
-
WHERE tenant_id = ${tenantId} AND tool_call_id = ${callId}
|
|
7788
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND tool_call_id = ${callId}
|
|
7788
7789
|
`;
|
|
7789
7790
|
const insertCallBody = (tenantId, input, definition, placement = { kind: "local" }) => sql`
|
|
7790
7791
|
(
|
|
@@ -7823,13 +7824,13 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7823
7824
|
`,
|
|
7824
7825
|
orElse: () => (tenantId, input, definition, placement) => sql`
|
|
7825
7826
|
INSERT INTO relay_tool_calls ${insertCallBody(tenantId, input, definition, placement)}
|
|
7826
|
-
ON CONFLICT (tenant_id, id) DO NOTHING
|
|
7827
|
+
ON CONFLICT (tenant_id, execution_id, id) DO NOTHING
|
|
7827
7828
|
`
|
|
7828
7829
|
});
|
|
7829
7830
|
const insertCall = sql.onDialectOrElse({
|
|
7830
7831
|
mysql: () => (tenantId, input, definition, placement) => sql.withTransaction(Effect37.gen(function* () {
|
|
7831
7832
|
yield* insertCallStatement(tenantId, input, definition, placement);
|
|
7832
|
-
return yield* selectCallByKey(tenantId, input.call.id);
|
|
7833
|
+
return yield* selectCallByKey(tenantId, input.executionId, input.call.id);
|
|
7833
7834
|
})),
|
|
7834
7835
|
orElse: () => (tenantId, input, definition, placement) => sql`
|
|
7835
7836
|
${insertCallStatement(tenantId, input, definition, placement)}
|
|
@@ -7838,10 +7839,11 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7838
7839
|
});
|
|
7839
7840
|
const insertResultStatement = (tenantId, input) => sql`
|
|
7840
7841
|
INSERT INTO relay_tool_results (
|
|
7841
|
-
tenant_id, ${resultColumns}
|
|
7842
|
+
tenant_id, execution_id, ${resultColumns}
|
|
7842
7843
|
)
|
|
7843
7844
|
VALUES (
|
|
7844
7845
|
${tenantId},
|
|
7846
|
+
${input.executionId},
|
|
7845
7847
|
${input.result.call_id},
|
|
7846
7848
|
${encodeJson(input.result.output)},
|
|
7847
7849
|
${input.result.error ?? null},
|
|
@@ -7849,19 +7851,19 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7849
7851
|
${timestampParam(sql, input.createdAt)}
|
|
7850
7852
|
)
|
|
7851
7853
|
`;
|
|
7852
|
-
const selectAttempts = (tenantId, callId) => sql`
|
|
7854
|
+
const selectAttempts = (tenantId, executionId, callId) => sql`
|
|
7853
7855
|
SELECT ${attemptColumns}
|
|
7854
7856
|
FROM relay_tool_attempts
|
|
7855
|
-
WHERE tenant_id = ${tenantId} AND tool_call_id = ${callId}
|
|
7857
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND tool_call_id = ${callId}
|
|
7856
7858
|
ORDER BY attempt_number ASC
|
|
7857
7859
|
`;
|
|
7858
7860
|
const lockedCallRows = sql.onDialectOrElse({
|
|
7859
|
-
pg: () => (tenantId, callId) => sql`
|
|
7860
|
-
${selectCallByKey(tenantId, callId)}
|
|
7861
|
+
pg: () => (tenantId, executionId, callId) => sql`
|
|
7862
|
+
${selectCallByKey(tenantId, executionId, callId)}
|
|
7861
7863
|
FOR UPDATE
|
|
7862
7864
|
`,
|
|
7863
|
-
mysql: () => (tenantId, callId) => sql`
|
|
7864
|
-
${selectCallByKey(tenantId, callId)}
|
|
7865
|
+
mysql: () => (tenantId, executionId, callId) => sql`
|
|
7866
|
+
${selectCallByKey(tenantId, executionId, callId)}
|
|
7865
7867
|
FOR UPDATE
|
|
7866
7868
|
`,
|
|
7867
7869
|
orElse: () => selectCallByKey
|
|
@@ -7874,35 +7876,35 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7874
7876
|
const decoded = yield* decodeAttemptRow(row);
|
|
7875
7877
|
return yield* Effect37.try({ try: () => toAttemptRecord(decoded), catch: toRepositoryError17 });
|
|
7876
7878
|
});
|
|
7877
|
-
const readLockedCall = Effect37.fn("ToolCallRepository.readLockedCall")(function* (tenantId, callId) {
|
|
7878
|
-
const rows = yield* lockedCallRows(tenantId, callId).pipe(Effect37.mapError(toRepositoryError17));
|
|
7879
|
+
const readLockedCall = Effect37.fn("ToolCallRepository.readLockedCall")(function* (tenantId, executionId, callId) {
|
|
7880
|
+
const rows = yield* lockedCallRows(tenantId, executionId, callId).pipe(Effect37.mapError(toRepositoryError17));
|
|
7879
7881
|
if (rows[0] === undefined)
|
|
7880
7882
|
return yield* ToolCallNotFound.make({ id: callId });
|
|
7881
7883
|
return yield* decodeCall(rows[0]);
|
|
7882
7884
|
});
|
|
7883
|
-
const readAttempts = Effect37.fn("ToolCallRepository.readAttempts")(function* (tenantId, callId) {
|
|
7884
|
-
const rows = yield* selectAttempts(tenantId, callId).pipe(Effect37.mapError(toRepositoryError17));
|
|
7885
|
+
const readAttempts = Effect37.fn("ToolCallRepository.readAttempts")(function* (tenantId, executionId, callId) {
|
|
7886
|
+
const rows = yield* selectAttempts(tenantId, executionId, callId).pipe(Effect37.mapError(toRepositoryError17));
|
|
7885
7887
|
return yield* Effect37.forEach(rows, decodeAttempt);
|
|
7886
7888
|
});
|
|
7887
|
-
const nextAttemptNumber = Effect37.fn("ToolCallRepository.nextAttemptNumber")(function* (tenantId, callId) {
|
|
7889
|
+
const nextAttemptNumber = Effect37.fn("ToolCallRepository.nextAttemptNumber")(function* (tenantId, executionId, callId) {
|
|
7888
7890
|
const rows = yield* sql`
|
|
7889
7891
|
SELECT COALESCE(MAX(attempt_number), 0) AS max_attempt
|
|
7890
7892
|
FROM relay_tool_attempts
|
|
7891
|
-
WHERE tenant_id = ${tenantId} AND tool_call_id = ${callId}
|
|
7893
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND tool_call_id = ${callId}
|
|
7892
7894
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
7893
7895
|
const row = yield* decodeMaxAttemptRow(rows[0]);
|
|
7894
7896
|
const max = Number(row?.max_attempt ?? 0);
|
|
7895
7897
|
return max + 1;
|
|
7896
7898
|
});
|
|
7897
|
-
const insertAttemptNumber = Effect37.fn("ToolCallRepository.insertAttemptNumber")(function* (tenantId, callId, attemptNumber, createdAt, workerId, claimExpiresAt) {
|
|
7899
|
+
const insertAttemptNumber = Effect37.fn("ToolCallRepository.insertAttemptNumber")(function* (tenantId, executionId, callId, attemptNumber, createdAt, workerId, claimExpiresAt) {
|
|
7898
7900
|
const attemptId = toolAttemptId(attemptNumber);
|
|
7899
7901
|
yield* sql`
|
|
7900
7902
|
INSERT INTO relay_tool_attempts (
|
|
7901
|
-
tenant_id, id, tool_call_id, attempt_number, state, worker_id, claim_expires_at,
|
|
7903
|
+
tenant_id, execution_id, id, tool_call_id, attempt_number, state, worker_id, claim_expires_at,
|
|
7902
7904
|
error, created_at, completed_at
|
|
7903
7905
|
)
|
|
7904
7906
|
VALUES (
|
|
7905
|
-
${tenantId}, ${attemptId}, ${callId}, ${attemptNumber}, 'running', ${workerId ?? null},
|
|
7907
|
+
${tenantId}, ${executionId}, ${attemptId}, ${callId}, ${attemptNumber}, 'running', ${workerId ?? null},
|
|
7906
7908
|
${claimExpiresAt === undefined ? null : ts(claimExpiresAt)}, ${null}, ${ts(createdAt)}, ${null}
|
|
7907
7909
|
)
|
|
7908
7910
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
@@ -7916,25 +7918,25 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7916
7918
|
createdAt
|
|
7917
7919
|
};
|
|
7918
7920
|
});
|
|
7919
|
-
const insertAttempt = Effect37.fn("ToolCallRepository.insertAttempt")(function* (tenantId, callId, createdAt, workerId, claimExpiresAt) {
|
|
7920
|
-
const attemptNumber = yield* nextAttemptNumber(tenantId, callId);
|
|
7921
|
-
return yield* insertAttemptNumber(tenantId, callId, attemptNumber, createdAt, workerId, claimExpiresAt);
|
|
7921
|
+
const insertAttempt = Effect37.fn("ToolCallRepository.insertAttempt")(function* (tenantId, executionId, callId, createdAt, workerId, claimExpiresAt) {
|
|
7922
|
+
const attemptNumber = yield* nextAttemptNumber(tenantId, executionId, callId);
|
|
7923
|
+
return yield* insertAttemptNumber(tenantId, executionId, callId, attemptNumber, createdAt, workerId, claimExpiresAt);
|
|
7922
7924
|
});
|
|
7923
|
-
const loadAttempt = Effect37.fn("ToolCallRepository.loadAttempt")(function* (tenantId, callId, attemptId) {
|
|
7925
|
+
const loadAttempt = Effect37.fn("ToolCallRepository.loadAttempt")(function* (tenantId, executionId, callId, attemptId) {
|
|
7924
7926
|
const rows = yield* sql`
|
|
7925
7927
|
SELECT ${attemptColumns}
|
|
7926
7928
|
FROM relay_tool_attempts
|
|
7927
|
-
WHERE tenant_id = ${tenantId} AND tool_call_id = ${callId} AND id = ${attemptId}
|
|
7929
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND tool_call_id = ${callId} AND id = ${attemptId}
|
|
7928
7930
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
7929
7931
|
return rows[0] === undefined ? undefined : yield* decodeAttempt(rows[0]);
|
|
7930
7932
|
});
|
|
7931
|
-
const completeAttempt = Effect37.fn("ToolCallRepository.completeAttempt")(function* (tenantId, callId, attemptId, state, completedAt, error5) {
|
|
7933
|
+
const completeAttempt = Effect37.fn("ToolCallRepository.completeAttempt")(function* (tenantId, executionId, callId, attemptId, state, completedAt, error5) {
|
|
7932
7934
|
yield* sql`
|
|
7933
7935
|
UPDATE relay_tool_attempts
|
|
7934
7936
|
SET state = ${state}, error = ${error5 ?? null}, completed_at = ${ts(completedAt)}
|
|
7935
|
-
WHERE tenant_id = ${tenantId} AND tool_call_id = ${callId} AND id = ${attemptId} AND state = 'running'
|
|
7937
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND tool_call_id = ${callId} AND id = ${attemptId} AND state = 'running'
|
|
7936
7938
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
7937
|
-
const attempt = yield* loadAttempt(tenantId, callId, attemptId);
|
|
7939
|
+
const attempt = yield* loadAttempt(tenantId, executionId, callId, attemptId);
|
|
7938
7940
|
if (attempt === undefined) {
|
|
7939
7941
|
return yield* ToolCallRepositoryError.make({ message: `Tool attempt not found: ${attemptId}` });
|
|
7940
7942
|
}
|
|
@@ -7944,7 +7946,7 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7944
7946
|
const tenantId = yield* current();
|
|
7945
7947
|
return yield* sql.withTransaction(Effect37.gen(function* () {
|
|
7946
7948
|
yield* insertCallIgnore(tenantId, input, input.definition, input.placement);
|
|
7947
|
-
const rows = yield* lockedCallRows(tenantId, input.call.id);
|
|
7949
|
+
const rows = yield* lockedCallRows(tenantId, input.executionId, input.call.id);
|
|
7948
7950
|
if (rows[0] === undefined)
|
|
7949
7951
|
return yield* ToolCallNotFound.make({ id: input.call.id });
|
|
7950
7952
|
const existing = yield* decodeCall(rows[0]);
|
|
@@ -7960,7 +7962,7 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7960
7962
|
const prepareExternalWait = Effect37.fn("ToolCallRepository.prepareExternalWait")(function* (input) {
|
|
7961
7963
|
const tenantId = yield* current();
|
|
7962
7964
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
7963
|
-
const call = yield* readLockedCall(tenantId, input.callId);
|
|
7965
|
+
const call = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
7964
7966
|
if (call.placement.kind !== "client" && call.placement.kind !== "remote") {
|
|
7965
7967
|
return yield* ToolCallTransitionRejected.make({
|
|
7966
7968
|
id: input.callId,
|
|
@@ -7976,15 +7978,15 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7976
7978
|
UPDATE relay_tool_calls
|
|
7977
7979
|
SET state = 'waiting', wait_id = ${input.waitId}, available_at = ${ts(input.availableAt)},
|
|
7978
7980
|
updated_at = ${ts(input.updatedAt)}
|
|
7979
|
-
WHERE tenant_id = ${tenantId} AND id = ${input.callId} AND state = 'requested'
|
|
7981
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${input.executionId} AND id = ${input.callId} AND state = 'requested'
|
|
7980
7982
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
7981
|
-
return yield* readLockedCall(tenantId, input.callId);
|
|
7983
|
+
return yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
7982
7984
|
}));
|
|
7983
7985
|
});
|
|
7984
7986
|
const beginAttempt = Effect37.fn("ToolCallRepository.beginAttempt")(function* (input) {
|
|
7985
7987
|
const tenantId = yield* current();
|
|
7986
7988
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
7987
|
-
const call = yield* readLockedCall(tenantId, input.callId);
|
|
7989
|
+
const call = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
7988
7990
|
if (call.placement.kind !== "local" && call.placement.kind !== "mcp") {
|
|
7989
7991
|
return yield* ToolCallTransitionRejected.make({
|
|
7990
7992
|
id: input.callId,
|
|
@@ -7995,13 +7997,13 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
7995
7997
|
return yield* ToolCallTransitionRejected.make({ id: input.callId, message: `Call is ${call.state}` });
|
|
7996
7998
|
}
|
|
7997
7999
|
if (call.activeAttemptId !== undefined) {
|
|
7998
|
-
yield* completeAttempt(tenantId, input.callId, call.activeAttemptId, "abandoned", input.startedAt);
|
|
8000
|
+
yield* completeAttempt(tenantId, input.executionId, input.callId, call.activeAttemptId, "abandoned", input.startedAt);
|
|
7999
8001
|
}
|
|
8000
|
-
const attempt = yield* insertAttempt(tenantId, input.callId, input.startedAt);
|
|
8002
|
+
const attempt = yield* insertAttempt(tenantId, input.executionId, input.callId, input.startedAt);
|
|
8001
8003
|
yield* sql`
|
|
8002
8004
|
UPDATE relay_tool_calls
|
|
8003
8005
|
SET state = 'running', active_attempt_id = ${attempt.id}, updated_at = ${ts(input.startedAt)}
|
|
8004
|
-
WHERE tenant_id = ${tenantId} AND id = ${input.callId}
|
|
8006
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${input.executionId} AND id = ${input.callId}
|
|
8005
8007
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8006
8008
|
return attempt;
|
|
8007
8009
|
}));
|
|
@@ -8009,7 +8011,7 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8009
8011
|
const acceptClientOutcome = Effect37.fn("ToolCallRepository.acceptClientOutcome")(function* (input) {
|
|
8010
8012
|
const tenantId = yield* current();
|
|
8011
8013
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
8012
|
-
const call = yield* readLockedCall(tenantId, input.callId);
|
|
8014
|
+
const call = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
8013
8015
|
if (call.executionId !== input.executionId || call.placement.kind !== "client") {
|
|
8014
8016
|
return yield* ToolCallTransitionRejected.make({
|
|
8015
8017
|
id: input.callId,
|
|
@@ -8026,7 +8028,7 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8026
8028
|
message: "Accepted attempt is missing"
|
|
8027
8029
|
});
|
|
8028
8030
|
}
|
|
8029
|
-
const attempt2 = yield* loadAttempt(tenantId, input.callId, call.activeAttemptId);
|
|
8031
|
+
const attempt2 = yield* loadAttempt(tenantId, input.executionId, input.callId, call.activeAttemptId);
|
|
8030
8032
|
if (attempt2 === undefined) {
|
|
8031
8033
|
return yield* ToolCallTransitionRejected.make({
|
|
8032
8034
|
id: input.callId,
|
|
@@ -8038,21 +8040,21 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8038
8040
|
if (call.state !== "waiting") {
|
|
8039
8041
|
return yield* ToolCallTransitionRejected.make({ id: input.callId, message: `Call is ${call.state}` });
|
|
8040
8042
|
}
|
|
8041
|
-
const running = yield* insertAttempt(tenantId, input.callId, input.submittedAt);
|
|
8042
|
-
const attempt = yield* completeAttempt(tenantId, input.callId, running.id, input.outcome.kind === "success" ? "completed" : "failed", input.submittedAt, input.outcome.kind === "failure" ? input.outcome.message : undefined);
|
|
8043
|
+
const running = yield* insertAttempt(tenantId, input.executionId, input.callId, input.submittedAt);
|
|
8044
|
+
const attempt = yield* completeAttempt(tenantId, input.executionId, input.callId, running.id, input.outcome.kind === "success" ? "completed" : "failed", input.submittedAt, input.outcome.kind === "failure" ? input.outcome.message : undefined);
|
|
8043
8045
|
yield* sql`
|
|
8044
8046
|
UPDATE relay_tool_calls
|
|
8045
8047
|
SET state = 'submitted', active_attempt_id = ${attempt.id},
|
|
8046
8048
|
external_outcome_json = ${encodeJson(input.outcome)},
|
|
8047
8049
|
external_outcome_at = ${ts(input.submittedAt)}, updated_at = ${ts(input.submittedAt)}
|
|
8048
|
-
WHERE tenant_id = ${tenantId} AND id = ${input.callId}
|
|
8050
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${input.executionId} AND id = ${input.callId}
|
|
8049
8051
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8050
|
-
const updated = yield* readLockedCall(tenantId, input.callId);
|
|
8052
|
+
const updated = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
8051
8053
|
return { call: updated, attempt, duplicate: false };
|
|
8052
8054
|
}));
|
|
8053
8055
|
});
|
|
8054
8056
|
const claimCandidate = (tenantId, input) => sql`
|
|
8055
|
-
SELECT id
|
|
8057
|
+
SELECT execution_id, id
|
|
8056
8058
|
FROM relay_tool_calls
|
|
8057
8059
|
WHERE tenant_id = ${tenantId} AND placement_kind = 'remote' AND placement_key = ${input.queue}
|
|
8058
8060
|
AND state = 'waiting' AND available_at <= ${ts(input.now)}
|
|
@@ -8071,20 +8073,20 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8071
8073
|
orElse: () => claimCandidate
|
|
8072
8074
|
});
|
|
8073
8075
|
const claimCall = sql.onDialectOrElse({
|
|
8074
|
-
mysql: () => (tenantId, callId, attemptId, input) => sql`
|
|
8076
|
+
mysql: () => (tenantId, executionId, callId, attemptId, input) => sql`
|
|
8075
8077
|
UPDATE relay_tool_calls
|
|
8076
8078
|
SET state = 'running', active_attempt_id = ${attemptId}, claim_owner = ${input.workerId},
|
|
8077
8079
|
claimed_at = ${ts(input.now)}, claim_expires_at = ${ts(input.claimExpiresAt)},
|
|
8078
8080
|
updated_at = ${ts(input.now)}
|
|
8079
|
-
WHERE tenant_id = ${tenantId} AND id = ${callId} AND state = 'waiting'
|
|
8081
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND id = ${callId} AND state = 'waiting'
|
|
8080
8082
|
AND available_at <= ${ts(input.now)}
|
|
8081
8083
|
`.pipe(Effect37.as([{ id: callId }])),
|
|
8082
|
-
orElse: () => (tenantId, callId, attemptId, input) => sql`
|
|
8084
|
+
orElse: () => (tenantId, executionId, callId, attemptId, input) => sql`
|
|
8083
8085
|
UPDATE relay_tool_calls
|
|
8084
8086
|
SET state = 'running', active_attempt_id = ${attemptId}, claim_owner = ${input.workerId},
|
|
8085
8087
|
claimed_at = ${ts(input.now)}, claim_expires_at = ${ts(input.claimExpiresAt)},
|
|
8086
8088
|
updated_at = ${ts(input.now)}
|
|
8087
|
-
WHERE tenant_id = ${tenantId} AND id = ${callId} AND state = 'waiting'
|
|
8089
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND id = ${callId} AND state = 'waiting'
|
|
8088
8090
|
AND available_at <= ${ts(input.now)}
|
|
8089
8091
|
RETURNING id
|
|
8090
8092
|
`
|
|
@@ -8096,21 +8098,22 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8096
8098
|
const tenantId = yield* current();
|
|
8097
8099
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
8098
8100
|
const expiredRows = yield* sql`
|
|
8099
|
-
SELECT id, active_attempt_id
|
|
8101
|
+
SELECT execution_id, id, active_attempt_id
|
|
8100
8102
|
FROM relay_tool_calls
|
|
8101
8103
|
WHERE tenant_id = ${tenantId} AND placement_kind = 'remote' AND placement_key = ${input.queue}
|
|
8102
8104
|
AND state = 'running' AND claim_expires_at IS NOT NULL AND claim_expires_at <= ${ts(input.now)}
|
|
8103
8105
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8104
8106
|
const expired = yield* Schema35.decodeUnknownEffect(Schema35.Array(ExpiredCallRow))(expiredRows).pipe(Effect37.mapError(toRepositoryError17));
|
|
8105
8107
|
for (const row of expired) {
|
|
8108
|
+
const executionId2 = exports_ids_schema.ExecutionId.make(row.execution_id);
|
|
8106
8109
|
if (row.active_attempt_id !== null && row.active_attempt_id !== undefined) {
|
|
8107
|
-
yield* completeAttempt(tenantId, exports_ids_schema.ToolCallId.make(row.id), exports_ids_schema.ToolAttemptId.make(row.active_attempt_id), "expired", input.now);
|
|
8110
|
+
yield* completeAttempt(tenantId, executionId2, exports_ids_schema.ToolCallId.make(row.id), exports_ids_schema.ToolAttemptId.make(row.active_attempt_id), "expired", input.now);
|
|
8108
8111
|
}
|
|
8109
8112
|
yield* sql`
|
|
8110
8113
|
UPDATE relay_tool_calls
|
|
8111
8114
|
SET state = 'waiting', active_attempt_id = NULL, claim_owner = NULL, claimed_at = NULL,
|
|
8112
8115
|
claim_expires_at = NULL, updated_at = ${ts(input.now)}
|
|
8113
|
-
WHERE tenant_id = ${tenantId} AND id = ${row.id} AND state = 'running'
|
|
8116
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId2} AND id = ${row.id} AND state = 'running'
|
|
8114
8117
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8115
8118
|
}
|
|
8116
8119
|
const candidateRows = yield* selectClaimCandidate(tenantId, input).pipe(Effect37.mapError(toRepositoryError17));
|
|
@@ -8118,21 +8121,22 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8118
8121
|
const candidate = candidates[0];
|
|
8119
8122
|
if (candidate === undefined)
|
|
8120
8123
|
return;
|
|
8124
|
+
const executionId = exports_ids_schema.ExecutionId.make(candidate.execution_id);
|
|
8121
8125
|
const callId = exports_ids_schema.ToolCallId.make(candidate.id);
|
|
8122
|
-
const call = yield* readLockedCall(tenantId, callId);
|
|
8126
|
+
const call = yield* readLockedCall(tenantId, executionId, callId);
|
|
8123
8127
|
if (call.definition === undefined || call.placement.kind !== "remote") {
|
|
8124
8128
|
return yield* ToolCallTransitionRejected.make({
|
|
8125
8129
|
id: callId,
|
|
8126
8130
|
message: "Remote call definition snapshot is missing"
|
|
8127
8131
|
});
|
|
8128
8132
|
}
|
|
8129
|
-
const attemptNumber = yield* nextAttemptNumber(tenantId, callId);
|
|
8133
|
+
const attemptNumber = yield* nextAttemptNumber(tenantId, executionId, callId);
|
|
8130
8134
|
const attemptId = toolAttemptId(attemptNumber);
|
|
8131
|
-
const claimed = yield* claimCall(tenantId, callId, attemptId, input).pipe(Effect37.mapError(toRepositoryError17));
|
|
8135
|
+
const claimed = yield* claimCall(tenantId, executionId, callId, attemptId, input).pipe(Effect37.mapError(toRepositoryError17));
|
|
8132
8136
|
if (claimed[0] === undefined)
|
|
8133
8137
|
return;
|
|
8134
|
-
const attempt = yield* insertAttemptNumber(tenantId, callId, attemptNumber, input.now, input.workerId, input.claimExpiresAt);
|
|
8135
|
-
const updated = yield* readLockedCall(tenantId, callId);
|
|
8138
|
+
const attempt = yield* insertAttemptNumber(tenantId, executionId, callId, attemptNumber, input.now, input.workerId, input.claimExpiresAt);
|
|
8139
|
+
const updated = yield* readLockedCall(tenantId, executionId, callId);
|
|
8136
8140
|
if (updated.definition === undefined || updated.placement.kind !== "remote") {
|
|
8137
8141
|
return yield* ToolCallTransitionRejected.make({
|
|
8138
8142
|
id: callId,
|
|
@@ -8152,25 +8156,25 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8152
8156
|
const releaseRemote = Effect37.fn("ToolCallRepository.releaseRemote")(function* (input) {
|
|
8153
8157
|
const tenantId = yield* current();
|
|
8154
8158
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
8155
|
-
const call = yield* readLockedCall(tenantId, input.callId);
|
|
8159
|
+
const call = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
8156
8160
|
if (call.placement.kind !== "remote" || call.state !== "running" || call.activeAttemptId !== input.attemptId || call.claimOwner !== input.workerId || call.claimExpiresAt === undefined || call.claimExpiresAt <= input.releasedAt) {
|
|
8157
8161
|
return yield* ToolClaimRejected.make({ message: "Remote claim is not active" });
|
|
8158
8162
|
}
|
|
8159
|
-
yield* completeAttempt(tenantId, input.callId, input.attemptId, "released", input.releasedAt, input.error);
|
|
8163
|
+
yield* completeAttempt(tenantId, input.executionId, input.callId, input.attemptId, "released", input.releasedAt, input.error);
|
|
8160
8164
|
yield* sql`
|
|
8161
8165
|
UPDATE relay_tool_calls
|
|
8162
8166
|
SET state = 'waiting', available_at = ${ts(input.nextAvailableAt)}, active_attempt_id = NULL,
|
|
8163
8167
|
claim_owner = NULL, claimed_at = NULL, claim_expires_at = NULL,
|
|
8164
8168
|
updated_at = ${ts(input.releasedAt)}
|
|
8165
|
-
WHERE tenant_id = ${tenantId} AND id = ${input.callId}
|
|
8169
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${input.executionId} AND id = ${input.callId}
|
|
8166
8170
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8167
|
-
return yield* readLockedCall(tenantId, input.callId);
|
|
8171
|
+
return yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
8168
8172
|
}));
|
|
8169
8173
|
});
|
|
8170
8174
|
const acceptRemoteOutcome = Effect37.fn("ToolCallRepository.acceptRemoteOutcome")(function* (input) {
|
|
8171
8175
|
const tenantId = yield* current();
|
|
8172
8176
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
8173
|
-
const call = yield* readLockedCall(tenantId, input.callId);
|
|
8177
|
+
const call = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
8174
8178
|
if (call.externalOutcome !== undefined) {
|
|
8175
8179
|
if (!exports_shared_schema.canonicalEquals(call.externalOutcome, input.outcome)) {
|
|
8176
8180
|
return yield* ToolOutcomeConflict.make({ id: input.callId });
|
|
@@ -8178,7 +8182,7 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8178
8182
|
if (call.activeAttemptId !== input.attemptId) {
|
|
8179
8183
|
return yield* ToolClaimRejected.make({ message: "Remote attempt was not accepted" });
|
|
8180
8184
|
}
|
|
8181
|
-
const attempt2 = yield* loadAttempt(tenantId, input.callId, input.attemptId);
|
|
8185
|
+
const attempt2 = yield* loadAttempt(tenantId, input.executionId, input.callId, input.attemptId);
|
|
8182
8186
|
if (attempt2 === undefined || attempt2.callId !== input.callId || attempt2.workerId !== input.workerId) {
|
|
8183
8187
|
return yield* ToolClaimRejected.make({ message: "Remote attempt was not accepted" });
|
|
8184
8188
|
}
|
|
@@ -8187,21 +8191,21 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8187
8191
|
if (call.placement.kind !== "remote" || call.state !== "running" || call.activeAttemptId !== input.attemptId || call.claimOwner !== input.workerId || call.claimExpiresAt === undefined || call.claimExpiresAt <= input.submittedAt) {
|
|
8188
8192
|
return yield* ToolClaimRejected.make({ message: "Remote claim is not active" });
|
|
8189
8193
|
}
|
|
8190
|
-
const attempt = yield* completeAttempt(tenantId, input.callId, input.attemptId, input.outcome.kind === "success" ? "completed" : "failed", input.submittedAt, input.outcome.kind === "failure" ? input.outcome.message : undefined);
|
|
8194
|
+
const attempt = yield* completeAttempt(tenantId, input.executionId, input.callId, input.attemptId, input.outcome.kind === "success" ? "completed" : "failed", input.submittedAt, input.outcome.kind === "failure" ? input.outcome.message : undefined);
|
|
8191
8195
|
yield* sql`
|
|
8192
8196
|
UPDATE relay_tool_calls
|
|
8193
8197
|
SET state = 'submitted', external_outcome_json = ${encodeJson(input.outcome)},
|
|
8194
8198
|
external_outcome_at = ${ts(input.submittedAt)}, claim_owner = NULL, claimed_at = NULL,
|
|
8195
8199
|
claim_expires_at = NULL, updated_at = ${ts(input.submittedAt)}
|
|
8196
|
-
WHERE tenant_id = ${tenantId} AND id = ${input.callId}
|
|
8200
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${input.executionId} AND id = ${input.callId}
|
|
8197
8201
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8198
|
-
const updated = yield* readLockedCall(tenantId, input.callId);
|
|
8202
|
+
const updated = yield* readLockedCall(tenantId, input.executionId, input.callId);
|
|
8199
8203
|
return { call: updated, attempt, duplicate: false };
|
|
8200
8204
|
}));
|
|
8201
8205
|
});
|
|
8202
|
-
const listAttempts = Effect37.fn("ToolCallRepository.listAttempts")(function* (callId) {
|
|
8206
|
+
const listAttempts = Effect37.fn("ToolCallRepository.listAttempts")(function* (executionId, callId) {
|
|
8203
8207
|
const tenantId = yield* current();
|
|
8204
|
-
return yield* readAttempts(tenantId, callId);
|
|
8208
|
+
return yield* readAttempts(tenantId, executionId, callId);
|
|
8205
8209
|
});
|
|
8206
8210
|
const selectOpenCalls = sql.onDialectOrElse({
|
|
8207
8211
|
pg: () => (tenantId, input) => sql`
|
|
@@ -8229,17 +8233,17 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8229
8233
|
`
|
|
8230
8234
|
});
|
|
8231
8235
|
const failCall = sql.onDialectOrElse({
|
|
8232
|
-
mysql: () => (tenantId, callId, failedAt) => sql`
|
|
8236
|
+
mysql: () => (tenantId, executionId, callId, failedAt) => sql`
|
|
8233
8237
|
UPDATE relay_tool_calls
|
|
8234
8238
|
SET state = 'failed', claim_owner = NULL, claimed_at = NULL, claim_expires_at = NULL,
|
|
8235
8239
|
updated_at = ${ts(failedAt)}
|
|
8236
|
-
WHERE tenant_id = ${tenantId} AND id = ${callId} AND state NOT IN ('submitted', 'completed', 'failed')
|
|
8240
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND id = ${callId} AND state NOT IN ('submitted', 'completed', 'failed')
|
|
8237
8241
|
`.pipe(Effect37.as([{ id: callId }])),
|
|
8238
|
-
orElse: () => (tenantId, callId, failedAt) => sql`
|
|
8242
|
+
orElse: () => (tenantId, executionId, callId, failedAt) => sql`
|
|
8239
8243
|
UPDATE relay_tool_calls
|
|
8240
8244
|
SET state = 'failed', claim_owner = NULL, claimed_at = NULL, claim_expires_at = NULL,
|
|
8241
8245
|
updated_at = ${ts(failedAt)}
|
|
8242
|
-
WHERE tenant_id = ${tenantId} AND id = ${callId} AND state NOT IN ('submitted', 'completed', 'failed')
|
|
8246
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${executionId} AND id = ${callId} AND state NOT IN ('submitted', 'completed', 'failed')
|
|
8243
8247
|
RETURNING id
|
|
8244
8248
|
`
|
|
8245
8249
|
});
|
|
@@ -8253,30 +8257,31 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8253
8257
|
for (const call of open) {
|
|
8254
8258
|
if (selected !== undefined && !selected.has(call.call.id))
|
|
8255
8259
|
continue;
|
|
8256
|
-
const transitioned = yield* failCall(tenantId, call.call.id, input.failedAt).pipe(Effect37.mapError(toRepositoryError17));
|
|
8260
|
+
const transitioned = yield* failCall(tenantId, input.executionId, call.call.id, input.failedAt).pipe(Effect37.mapError(toRepositoryError17));
|
|
8257
8261
|
if (transitioned[0] === undefined)
|
|
8258
8262
|
continue;
|
|
8259
8263
|
if (call.activeAttemptId !== undefined) {
|
|
8260
|
-
const attempt = yield* loadAttempt(tenantId, call.call.id, call.activeAttemptId);
|
|
8264
|
+
const attempt = yield* loadAttempt(tenantId, input.executionId, call.call.id, call.activeAttemptId);
|
|
8261
8265
|
if (attempt?.state === "running") {
|
|
8262
|
-
yield* completeAttempt(tenantId, call.call.id, call.activeAttemptId, "failed", input.failedAt, input.message);
|
|
8266
|
+
yield* completeAttempt(tenantId, input.executionId, call.call.id, call.activeAttemptId, "failed", input.failedAt, input.message);
|
|
8263
8267
|
}
|
|
8264
8268
|
}
|
|
8265
|
-
const existingResult = yield* selectResultByKey(tenantId, call.call.id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8269
|
+
const existingResult = yield* selectResultByKey(tenantId, input.executionId, call.call.id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8266
8270
|
if (existingResult[0] === undefined) {
|
|
8267
8271
|
yield* insertResultStatement(tenantId, {
|
|
8272
|
+
executionId: input.executionId,
|
|
8268
8273
|
result: { call_id: call.call.id, output: null, error: input.message },
|
|
8269
8274
|
createdAt: input.failedAt
|
|
8270
8275
|
}).pipe(Effect37.mapError(toRepositoryError17));
|
|
8271
8276
|
}
|
|
8272
|
-
failed.push(yield* readLockedCall(tenantId, call.call.id));
|
|
8277
|
+
failed.push(yield* readLockedCall(tenantId, input.executionId, call.call.id));
|
|
8273
8278
|
}
|
|
8274
8279
|
return failed;
|
|
8275
8280
|
}));
|
|
8276
8281
|
});
|
|
8277
8282
|
const recordCall = Effect37.fn("ToolCallRepository.recordCall")(function* (input) {
|
|
8278
8283
|
const tenantId = yield* current();
|
|
8279
|
-
const existing = yield* selectCallByKey(tenantId, input.call.id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8284
|
+
const existing = yield* selectCallByKey(tenantId, input.executionId, input.call.id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8280
8285
|
if (existing[0] !== undefined)
|
|
8281
8286
|
return yield* DuplicateToolCall.make({ id: input.call.id });
|
|
8282
8287
|
const rows = yield* insertCall(tenantId, input).pipe(Effect37.mapError(toRepositoryError17));
|
|
@@ -8286,9 +8291,9 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8286
8291
|
}
|
|
8287
8292
|
return yield* decodeCall(row);
|
|
8288
8293
|
});
|
|
8289
|
-
const getCall = Effect37.fn("ToolCallRepository.getCall")(function* (id) {
|
|
8294
|
+
const getCall = Effect37.fn("ToolCallRepository.getCall")(function* (executionId, id) {
|
|
8290
8295
|
const tenantId = yield* current();
|
|
8291
|
-
const rows = yield* selectCallByKey(tenantId, id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8296
|
+
const rows = yield* selectCallByKey(tenantId, executionId, id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8292
8297
|
return rows[0] === undefined ? undefined : yield* decodeCall(rows[0]);
|
|
8293
8298
|
});
|
|
8294
8299
|
const listCalls = Effect37.fn("ToolCallRepository.listCalls")(function* (executionId) {
|
|
@@ -8304,16 +8309,16 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8304
8309
|
const recordResult = Effect37.fn("ToolCallRepository.recordResult")(function* (input) {
|
|
8305
8310
|
const tenantId = yield* current();
|
|
8306
8311
|
return yield* repositoryTransaction(Effect37.gen(function* () {
|
|
8307
|
-
const call = yield* readLockedCall(tenantId, input.result.call_id);
|
|
8308
|
-
const existing = yield* selectResultByKey(tenantId, input.result.call_id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8312
|
+
const call = yield* readLockedCall(tenantId, input.executionId, input.result.call_id);
|
|
8313
|
+
const existing = yield* selectResultByKey(tenantId, input.executionId, input.result.call_id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8309
8314
|
if (existing[0] !== undefined) {
|
|
8310
8315
|
return yield* DuplicateToolResult.make({ call_id: input.result.call_id });
|
|
8311
8316
|
}
|
|
8312
8317
|
yield* insertResultStatement(tenantId, input).pipe(Effect37.mapError(toRepositoryError17));
|
|
8313
8318
|
if (call.activeAttemptId !== undefined) {
|
|
8314
|
-
const attempt = yield* loadAttempt(tenantId, input.result.call_id, call.activeAttemptId);
|
|
8319
|
+
const attempt = yield* loadAttempt(tenantId, input.executionId, input.result.call_id, call.activeAttemptId);
|
|
8315
8320
|
if (attempt?.state === "running") {
|
|
8316
|
-
yield* completeAttempt(tenantId, input.result.call_id, call.activeAttemptId, input.result.error === undefined ? "completed" : "failed", input.createdAt, input.result.error);
|
|
8321
|
+
yield* completeAttempt(tenantId, input.executionId, input.result.call_id, call.activeAttemptId, input.result.error === undefined ? "completed" : "failed", input.createdAt, input.result.error);
|
|
8317
8322
|
}
|
|
8318
8323
|
}
|
|
8319
8324
|
yield* sql`
|
|
@@ -8321,18 +8326,18 @@ var layer31 = Layer36.effect(Service38, Effect37.gen(function* () {
|
|
|
8321
8326
|
SET state = ${input.result.error === undefined ? "completed" : "failed"},
|
|
8322
8327
|
claim_owner = NULL, claimed_at = NULL, claim_expires_at = NULL,
|
|
8323
8328
|
updated_at = ${ts(input.createdAt)}
|
|
8324
|
-
WHERE tenant_id = ${tenantId} AND id = ${input.result.call_id}
|
|
8329
|
+
WHERE tenant_id = ${tenantId} AND execution_id = ${input.executionId} AND id = ${input.result.call_id}
|
|
8325
8330
|
`.pipe(Effect37.mapError(toRepositoryError17));
|
|
8326
|
-
const rows = yield* selectResultByKey(tenantId, input.result.call_id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8331
|
+
const rows = yield* selectResultByKey(tenantId, input.executionId, input.result.call_id).pipe(Effect37.mapError(toRepositoryError17));
|
|
8327
8332
|
if (rows[0] === undefined) {
|
|
8328
8333
|
return yield* ToolCallRepositoryError.make({ message: "Tool result insert returned no row" });
|
|
8329
8334
|
}
|
|
8330
8335
|
return toResultRecord(yield* decodeResultRow(rows[0]));
|
|
8331
8336
|
}));
|
|
8332
8337
|
});
|
|
8333
|
-
const getResult = Effect37.fn("ToolCallRepository.getResult")(function* (callId) {
|
|
8338
|
+
const getResult = Effect37.fn("ToolCallRepository.getResult")(function* (executionId, callId) {
|
|
8334
8339
|
const tenantId = yield* current();
|
|
8335
|
-
const rows = yield* selectResultByKey(tenantId, callId).pipe(Effect37.mapError(toRepositoryError17));
|
|
8340
|
+
const rows = yield* selectResultByKey(tenantId, executionId, callId).pipe(Effect37.mapError(toRepositoryError17));
|
|
8336
8341
|
return rows[0] === undefined ? undefined : toResultRecord(yield* decodeResultRow(rows[0]));
|
|
8337
8342
|
});
|
|
8338
8343
|
return Service38.of({
|
|
@@ -8357,7 +8362,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8357
8362
|
const callTenants = new Map;
|
|
8358
8363
|
const results = new Map;
|
|
8359
8364
|
const attempts = new Map;
|
|
8360
|
-
const key3 = (tenantId, callId) => exports_shared_schema.canonicalString([tenantId, callId]);
|
|
8365
|
+
const key3 = (tenantId, executionId, callId) => exports_shared_schema.canonicalString([tenantId, executionId, callId]);
|
|
8361
8366
|
const same = (left, right) => exports_shared_schema.canonicalEquals(left, right);
|
|
8362
8367
|
const callRecord = (tenantId, input, definition, placement = { kind: "local" }) => ({
|
|
8363
8368
|
executionId: input.executionId,
|
|
@@ -8370,13 +8375,13 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8370
8375
|
createdAt: input.createdAt,
|
|
8371
8376
|
updatedAt: input.createdAt
|
|
8372
8377
|
});
|
|
8373
|
-
const attemptsFor = (tenantId, callId) => attempts.get(key3(tenantId, callId)) ?? [];
|
|
8374
|
-
const replaceAttempt = (tenantId, attempt) => {
|
|
8375
|
-
const callAttempts = attemptsFor(tenantId, attempt.callId);
|
|
8376
|
-
attempts.set(key3(tenantId, attempt.callId), callAttempts.map((existing) => existing.id === attempt.id ? attempt : existing));
|
|
8378
|
+
const attemptsFor = (tenantId, executionId, callId) => attempts.get(key3(tenantId, executionId, callId)) ?? [];
|
|
8379
|
+
const replaceAttempt = (tenantId, executionId, attempt) => {
|
|
8380
|
+
const callAttempts = attemptsFor(tenantId, executionId, attempt.callId);
|
|
8381
|
+
attempts.set(key3(tenantId, executionId, attempt.callId), callAttempts.map((existing) => existing.id === attempt.id ? attempt : existing));
|
|
8377
8382
|
};
|
|
8378
|
-
const appendAttempt = (tenantId, callId, createdAt, options = {}) => {
|
|
8379
|
-
const callAttempts = attemptsFor(tenantId, callId);
|
|
8383
|
+
const appendAttempt = (tenantId, executionId, callId, createdAt, options = {}) => {
|
|
8384
|
+
const callAttempts = attemptsFor(tenantId, executionId, callId);
|
|
8380
8385
|
const attemptNumber = callAttempts.length + 1;
|
|
8381
8386
|
const attempt = {
|
|
8382
8387
|
id: toolAttemptId(attemptNumber),
|
|
@@ -8387,18 +8392,18 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8387
8392
|
...options.claimExpiresAt === undefined ? {} : { claimExpiresAt: options.claimExpiresAt },
|
|
8388
8393
|
createdAt
|
|
8389
8394
|
};
|
|
8390
|
-
attempts.set(key3(tenantId, callId), [...callAttempts, attempt]);
|
|
8395
|
+
attempts.set(key3(tenantId, executionId, callId), [...callAttempts, attempt]);
|
|
8391
8396
|
return attempt;
|
|
8392
8397
|
};
|
|
8393
|
-
const requireCall = (tenantId, callId) => {
|
|
8394
|
-
const record2 = calls.get(key3(tenantId, callId));
|
|
8398
|
+
const requireCall = (tenantId, executionId, callId) => {
|
|
8399
|
+
const record2 = calls.get(key3(tenantId, executionId, callId));
|
|
8395
8400
|
return record2 === undefined ? Effect37.fail(ToolCallNotFound.make({ id: callId })) : Effect37.succeed(record2);
|
|
8396
8401
|
};
|
|
8397
8402
|
const transitionRejected = (id, message) => Effect37.fail(ToolCallTransitionRejected.make({ id, message }));
|
|
8398
8403
|
return Service38.of({
|
|
8399
8404
|
ensureCall: Effect37.fn("ToolCallRepository.memory.ensureCall")(function* (input) {
|
|
8400
8405
|
const tenantId = yield* current();
|
|
8401
|
-
const storageKey = key3(tenantId, input.call.id);
|
|
8406
|
+
const storageKey = key3(tenantId, input.executionId, input.call.id);
|
|
8402
8407
|
const existing = calls.get(storageKey);
|
|
8403
8408
|
if (existing !== undefined) {
|
|
8404
8409
|
if (existing.executionId !== input.executionId || !sameCallIdentity(existing.call, input.call) || !same(existing.definition, input.definition) || !same(existing.placement, input.placement)) {
|
|
@@ -8413,7 +8418,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8413
8418
|
}),
|
|
8414
8419
|
recordCall: Effect37.fn("ToolCallRepository.memory.recordCall")(function* (input) {
|
|
8415
8420
|
const tenantId = yield* current();
|
|
8416
|
-
const storageKey = key3(tenantId, input.call.id);
|
|
8421
|
+
const storageKey = key3(tenantId, input.executionId, input.call.id);
|
|
8417
8422
|
if (calls.has(storageKey))
|
|
8418
8423
|
return yield* DuplicateToolCall.make({ id: input.call.id });
|
|
8419
8424
|
const record2 = callRecord(tenantId, input);
|
|
@@ -8421,9 +8426,9 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8421
8426
|
callTenants.set(storageKey, tenantId);
|
|
8422
8427
|
return record2;
|
|
8423
8428
|
}),
|
|
8424
|
-
getCall: Effect37.fn("ToolCallRepository.memory.getCall")(function* (id) {
|
|
8429
|
+
getCall: Effect37.fn("ToolCallRepository.memory.getCall")(function* (executionId, id) {
|
|
8425
8430
|
const tenantId = yield* current();
|
|
8426
|
-
return calls.get(key3(tenantId, id));
|
|
8431
|
+
return calls.get(key3(tenantId, executionId, id));
|
|
8427
8432
|
}),
|
|
8428
8433
|
listCalls: Effect37.fn("ToolCallRepository.memory.listCalls")(function* (executionId) {
|
|
8429
8434
|
const tenantId = yield* current();
|
|
@@ -8431,7 +8436,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8431
8436
|
}),
|
|
8432
8437
|
recordResult: Effect37.fn("ToolCallRepository.memory.recordResult")(function* (input) {
|
|
8433
8438
|
const tenantId = yield* current();
|
|
8434
|
-
const storageKey = key3(tenantId, input.result.call_id);
|
|
8439
|
+
const storageKey = key3(tenantId, input.executionId, input.result.call_id);
|
|
8435
8440
|
const call = calls.get(storageKey);
|
|
8436
8441
|
if (call === undefined)
|
|
8437
8442
|
return yield* ToolCallNotFound.make({ id: input.result.call_id });
|
|
@@ -8439,9 +8444,9 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8439
8444
|
return yield* DuplicateToolResult.make({ call_id: input.result.call_id });
|
|
8440
8445
|
}
|
|
8441
8446
|
if (call.activeAttemptId !== undefined) {
|
|
8442
|
-
const active = attemptsFor(tenantId, input.result.call_id).find((attempt) => attempt.id === call.activeAttemptId);
|
|
8447
|
+
const active = attemptsFor(tenantId, input.executionId, input.result.call_id).find((attempt) => attempt.id === call.activeAttemptId);
|
|
8443
8448
|
if (active?.state === "running") {
|
|
8444
|
-
replaceAttempt(tenantId, {
|
|
8449
|
+
replaceAttempt(tenantId, input.executionId, {
|
|
8445
8450
|
...active,
|
|
8446
8451
|
state: input.result.error === undefined ? "completed" : "failed",
|
|
8447
8452
|
...input.result.error === undefined ? {} : { error: input.result.error },
|
|
@@ -8458,13 +8463,13 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8458
8463
|
});
|
|
8459
8464
|
return record2;
|
|
8460
8465
|
}),
|
|
8461
|
-
getResult: Effect37.fn("ToolCallRepository.memory.getResult")(function* (callId) {
|
|
8466
|
+
getResult: Effect37.fn("ToolCallRepository.memory.getResult")(function* (executionId, callId) {
|
|
8462
8467
|
const tenantId = yield* current();
|
|
8463
|
-
return results.get(key3(tenantId, callId));
|
|
8468
|
+
return results.get(key3(tenantId, executionId, callId));
|
|
8464
8469
|
}),
|
|
8465
8470
|
prepareExternalWait: Effect37.fn("ToolCallRepository.memory.prepareExternalWait")(function* (input) {
|
|
8466
8471
|
const tenantId = yield* current();
|
|
8467
|
-
const call = yield* requireCall(tenantId, input.callId);
|
|
8472
|
+
const call = yield* requireCall(tenantId, input.executionId, input.callId);
|
|
8468
8473
|
if (call.placement.kind !== "client" && call.placement.kind !== "remote") {
|
|
8469
8474
|
return yield* transitionRejected(input.callId, "Tool placement is not external");
|
|
8470
8475
|
}
|
|
@@ -8479,12 +8484,12 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8479
8484
|
availableAt: input.availableAt,
|
|
8480
8485
|
updatedAt: input.updatedAt
|
|
8481
8486
|
};
|
|
8482
|
-
calls.set(key3(tenantId, input.callId), updated);
|
|
8487
|
+
calls.set(key3(tenantId, input.executionId, input.callId), updated);
|
|
8483
8488
|
return updated;
|
|
8484
8489
|
}),
|
|
8485
8490
|
beginAttempt: Effect37.fn("ToolCallRepository.memory.beginAttempt")(function* (input) {
|
|
8486
8491
|
const tenantId = yield* current();
|
|
8487
|
-
const call = yield* requireCall(tenantId, input.callId);
|
|
8492
|
+
const call = yield* requireCall(tenantId, input.executionId, input.callId);
|
|
8488
8493
|
if (call.placement.kind !== "local" && call.placement.kind !== "mcp") {
|
|
8489
8494
|
return yield* transitionRejected(input.callId, "Tool placement is not immediate");
|
|
8490
8495
|
}
|
|
@@ -8492,13 +8497,13 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8492
8497
|
return yield* transitionRejected(input.callId, `Call is ${call.state}`);
|
|
8493
8498
|
}
|
|
8494
8499
|
if (call.activeAttemptId !== undefined) {
|
|
8495
|
-
const active = attemptsFor(tenantId, input.callId).find((attempt2) => attempt2.id === call.activeAttemptId);
|
|
8500
|
+
const active = attemptsFor(tenantId, input.executionId, input.callId).find((attempt2) => attempt2.id === call.activeAttemptId);
|
|
8496
8501
|
if (active?.state === "running") {
|
|
8497
|
-
replaceAttempt(tenantId, { ...active, state: "abandoned", completedAt: input.startedAt });
|
|
8502
|
+
replaceAttempt(tenantId, input.executionId, { ...active, state: "abandoned", completedAt: input.startedAt });
|
|
8498
8503
|
}
|
|
8499
8504
|
}
|
|
8500
|
-
const attempt = appendAttempt(tenantId, input.callId, input.startedAt);
|
|
8501
|
-
calls.set(key3(tenantId, input.callId), {
|
|
8505
|
+
const attempt = appendAttempt(tenantId, input.executionId, input.callId, input.startedAt);
|
|
8506
|
+
calls.set(key3(tenantId, input.executionId, input.callId), {
|
|
8502
8507
|
...call,
|
|
8503
8508
|
state: "running",
|
|
8504
8509
|
activeAttemptId: attempt.id,
|
|
@@ -8508,7 +8513,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8508
8513
|
}),
|
|
8509
8514
|
acceptClientOutcome: Effect37.fn("ToolCallRepository.memory.acceptClientOutcome")(function* (input) {
|
|
8510
8515
|
const tenantId = yield* current();
|
|
8511
|
-
const call = yield* requireCall(tenantId, input.callId);
|
|
8516
|
+
const call = yield* requireCall(tenantId, input.executionId, input.callId);
|
|
8512
8517
|
if (call.executionId !== input.executionId || call.placement.kind !== "client") {
|
|
8513
8518
|
return yield* transitionRejected(input.callId, "Client outcome does not match the call");
|
|
8514
8519
|
}
|
|
@@ -8516,21 +8521,21 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8516
8521
|
if (!same(call.externalOutcome, input.outcome)) {
|
|
8517
8522
|
return yield* ToolOutcomeConflict.make({ id: input.callId });
|
|
8518
8523
|
}
|
|
8519
|
-
const attempt2 = attemptsFor(tenantId, input.callId).find((item) => item.id === call.activeAttemptId);
|
|
8524
|
+
const attempt2 = attemptsFor(tenantId, input.executionId, input.callId).find((item) => item.id === call.activeAttemptId);
|
|
8520
8525
|
if (attempt2 === undefined)
|
|
8521
8526
|
return yield* transitionRejected(input.callId, "Accepted attempt is missing");
|
|
8522
8527
|
return { call, attempt: attempt2, duplicate: true };
|
|
8523
8528
|
}
|
|
8524
8529
|
if (call.state !== "waiting")
|
|
8525
8530
|
return yield* transitionRejected(input.callId, `Call is ${call.state}`);
|
|
8526
|
-
const running = appendAttempt(tenantId, input.callId, input.submittedAt);
|
|
8531
|
+
const running = appendAttempt(tenantId, input.executionId, input.callId, input.submittedAt);
|
|
8527
8532
|
const attempt = {
|
|
8528
8533
|
...running,
|
|
8529
8534
|
state: input.outcome.kind === "success" ? "completed" : "failed",
|
|
8530
8535
|
...input.outcome.kind === "failure" ? { error: input.outcome.message } : {},
|
|
8531
8536
|
completedAt: input.submittedAt
|
|
8532
8537
|
};
|
|
8533
|
-
replaceAttempt(tenantId, attempt);
|
|
8538
|
+
replaceAttempt(tenantId, input.executionId, attempt);
|
|
8534
8539
|
const updated = {
|
|
8535
8540
|
...call,
|
|
8536
8541
|
state: "submitted",
|
|
@@ -8539,7 +8544,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8539
8544
|
externalOutcomeAt: input.submittedAt,
|
|
8540
8545
|
updatedAt: input.submittedAt
|
|
8541
8546
|
};
|
|
8542
|
-
calls.set(key3(tenantId, input.callId), updated);
|
|
8547
|
+
calls.set(key3(tenantId, input.executionId, input.callId), updated);
|
|
8543
8548
|
return { call: updated, attempt, duplicate: false };
|
|
8544
8549
|
}),
|
|
8545
8550
|
claimRemote: Effect37.fn("ToolCallRepository.memory.claimRemote")(function* (input) {
|
|
@@ -8551,9 +8556,9 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8551
8556
|
if (callTenants.get(storageKey) !== tenantId || call.placement.kind !== "remote" || call.state !== "running" || call.claimExpiresAt === undefined || call.claimExpiresAt > input.now) {
|
|
8552
8557
|
continue;
|
|
8553
8558
|
}
|
|
8554
|
-
const active = attemptsFor(tenantId, call.call.id).find((attempt2) => attempt2.id === call.activeAttemptId);
|
|
8559
|
+
const active = attemptsFor(tenantId, call.executionId, call.call.id).find((attempt2) => attempt2.id === call.activeAttemptId);
|
|
8555
8560
|
if (active?.state === "running") {
|
|
8556
|
-
replaceAttempt(tenantId, { ...active, state: "expired", completedAt: input.now });
|
|
8561
|
+
replaceAttempt(tenantId, call.executionId, { ...active, state: "expired", completedAt: input.now });
|
|
8557
8562
|
}
|
|
8558
8563
|
calls.set(storageKey, {
|
|
8559
8564
|
...call,
|
|
@@ -8571,7 +8576,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8571
8576
|
if (candidate.definition === undefined) {
|
|
8572
8577
|
return yield* transitionRejected(candidate.call.id, "Remote call definition snapshot is missing");
|
|
8573
8578
|
}
|
|
8574
|
-
const attempt = appendAttempt(tenantId, candidate.call.id, input.now, {
|
|
8579
|
+
const attempt = appendAttempt(tenantId, candidate.executionId, candidate.call.id, input.now, {
|
|
8575
8580
|
workerId: input.workerId,
|
|
8576
8581
|
claimExpiresAt: input.claimExpiresAt
|
|
8577
8582
|
});
|
|
@@ -8585,7 +8590,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8585
8590
|
claimExpiresAt: input.claimExpiresAt,
|
|
8586
8591
|
updatedAt: input.now
|
|
8587
8592
|
};
|
|
8588
|
-
calls.set(key3(tenantId, candidate.call.id), updated);
|
|
8593
|
+
calls.set(key3(tenantId, candidate.executionId, candidate.call.id), updated);
|
|
8589
8594
|
return {
|
|
8590
8595
|
call: updated,
|
|
8591
8596
|
attempt,
|
|
@@ -8597,14 +8602,14 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8597
8602
|
}),
|
|
8598
8603
|
releaseRemote: Effect37.fn("ToolCallRepository.memory.releaseRemote")(function* (input) {
|
|
8599
8604
|
const tenantId = yield* current();
|
|
8600
|
-
const call = yield* requireCall(tenantId, input.callId);
|
|
8605
|
+
const call = yield* requireCall(tenantId, input.executionId, input.callId);
|
|
8601
8606
|
const valid = call.placement.kind === "remote" && call.state === "running" && call.activeAttemptId === input.attemptId && call.claimOwner === input.workerId && call.claimExpiresAt !== undefined && call.claimExpiresAt > input.releasedAt;
|
|
8602
8607
|
if (!valid)
|
|
8603
8608
|
return yield* ToolClaimRejected.make({ message: "Remote claim is not active" });
|
|
8604
|
-
const attempt = attemptsFor(tenantId, input.callId).find((item) => item.id === input.attemptId);
|
|
8609
|
+
const attempt = attemptsFor(tenantId, input.executionId, input.callId).find((item) => item.id === input.attemptId);
|
|
8605
8610
|
if (attempt === undefined)
|
|
8606
8611
|
return yield* transitionRejected(input.callId, "Remote attempt is missing");
|
|
8607
|
-
replaceAttempt(tenantId, {
|
|
8612
|
+
replaceAttempt(tenantId, input.executionId, {
|
|
8608
8613
|
...attempt,
|
|
8609
8614
|
state: "released",
|
|
8610
8615
|
...input.error === undefined ? {} : { error: input.error },
|
|
@@ -8620,12 +8625,12 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8620
8625
|
claimExpiresAt: undefined,
|
|
8621
8626
|
updatedAt: input.releasedAt
|
|
8622
8627
|
};
|
|
8623
|
-
calls.set(key3(tenantId, input.callId), updated);
|
|
8628
|
+
calls.set(key3(tenantId, input.executionId, input.callId), updated);
|
|
8624
8629
|
return updated;
|
|
8625
8630
|
}),
|
|
8626
8631
|
acceptRemoteOutcome: Effect37.fn("ToolCallRepository.memory.acceptRemoteOutcome")(function* (input) {
|
|
8627
8632
|
const tenantId = yield* current();
|
|
8628
|
-
const call = yield* requireCall(tenantId, input.callId);
|
|
8633
|
+
const call = yield* requireCall(tenantId, input.executionId, input.callId);
|
|
8629
8634
|
if (call.externalOutcome !== undefined) {
|
|
8630
8635
|
if (!same(call.externalOutcome, input.outcome)) {
|
|
8631
8636
|
return yield* ToolOutcomeConflict.make({ id: input.callId });
|
|
@@ -8633,7 +8638,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8633
8638
|
if (call.activeAttemptId !== input.attemptId) {
|
|
8634
8639
|
return yield* ToolClaimRejected.make({ message: "Remote attempt was not accepted" });
|
|
8635
8640
|
}
|
|
8636
|
-
const attempt2 = attemptsFor(tenantId, input.callId).find((item) => item.id === input.attemptId);
|
|
8641
|
+
const attempt2 = attemptsFor(tenantId, input.executionId, input.callId).find((item) => item.id === input.attemptId);
|
|
8637
8642
|
if (attempt2 === undefined || attempt2.callId !== input.callId || attempt2.workerId !== input.workerId) {
|
|
8638
8643
|
return yield* ToolClaimRejected.make({ message: "Remote attempt was not accepted" });
|
|
8639
8644
|
}
|
|
@@ -8642,7 +8647,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8642
8647
|
const valid = call.placement.kind === "remote" && call.state === "running" && call.activeAttemptId === input.attemptId && call.claimOwner === input.workerId && call.claimExpiresAt !== undefined && call.claimExpiresAt > input.submittedAt;
|
|
8643
8648
|
if (!valid)
|
|
8644
8649
|
return yield* ToolClaimRejected.make({ message: "Remote claim is not active" });
|
|
8645
|
-
const running = attemptsFor(tenantId, input.callId).find((item) => item.id === input.attemptId);
|
|
8650
|
+
const running = attemptsFor(tenantId, input.executionId, input.callId).find((item) => item.id === input.attemptId);
|
|
8646
8651
|
if (running === undefined)
|
|
8647
8652
|
return yield* transitionRejected(input.callId, "Remote attempt is missing");
|
|
8648
8653
|
const attempt = {
|
|
@@ -8651,7 +8656,7 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8651
8656
|
...input.outcome.kind === "failure" ? { error: input.outcome.message } : {},
|
|
8652
8657
|
completedAt: input.submittedAt
|
|
8653
8658
|
};
|
|
8654
|
-
replaceAttempt(tenantId, attempt);
|
|
8659
|
+
replaceAttempt(tenantId, input.executionId, attempt);
|
|
8655
8660
|
const updated = {
|
|
8656
8661
|
...call,
|
|
8657
8662
|
state: "submitted",
|
|
@@ -8662,12 +8667,12 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8662
8667
|
claimExpiresAt: undefined,
|
|
8663
8668
|
updatedAt: input.submittedAt
|
|
8664
8669
|
};
|
|
8665
|
-
calls.set(key3(tenantId, input.callId), updated);
|
|
8670
|
+
calls.set(key3(tenantId, input.executionId, input.callId), updated);
|
|
8666
8671
|
return { call: updated, attempt, duplicate: false };
|
|
8667
8672
|
}),
|
|
8668
|
-
listAttempts: Effect37.fn("ToolCallRepository.memory.listAttempts")(function* (callId) {
|
|
8673
|
+
listAttempts: Effect37.fn("ToolCallRepository.memory.listAttempts")(function* (executionId, callId) {
|
|
8669
8674
|
const tenantId = yield* current();
|
|
8670
|
-
return attemptsFor(tenantId, callId);
|
|
8675
|
+
return attemptsFor(tenantId, executionId, callId);
|
|
8671
8676
|
}),
|
|
8672
8677
|
failOpenForExecution: Effect37.fn("ToolCallRepository.memory.failOpenForExecution")(function* (input) {
|
|
8673
8678
|
const tenantId = yield* current();
|
|
@@ -8677,9 +8682,9 @@ var memoryLayer29 = Layer36.sync(Service38, () => {
|
|
|
8677
8682
|
continue;
|
|
8678
8683
|
}
|
|
8679
8684
|
if (call.activeAttemptId !== undefined) {
|
|
8680
|
-
const active = attemptsFor(tenantId, call.call.id).find((attempt) => attempt.id === call.activeAttemptId);
|
|
8685
|
+
const active = attemptsFor(tenantId, input.executionId, call.call.id).find((attempt) => attempt.id === call.activeAttemptId);
|
|
8681
8686
|
if (active?.state === "running") {
|
|
8682
|
-
replaceAttempt(tenantId, {
|
|
8687
|
+
replaceAttempt(tenantId, input.executionId, {
|
|
8683
8688
|
...active,
|
|
8684
8689
|
state: "failed",
|
|
8685
8690
|
error: input.message,
|
|
@@ -8723,9 +8728,9 @@ var recordCall = Effect37.fn("ToolCallRepository.recordCall.call")(function* (in
|
|
|
8723
8728
|
const repository = yield* Service38;
|
|
8724
8729
|
return yield* repository.recordCall(input);
|
|
8725
8730
|
});
|
|
8726
|
-
var getCall = Effect37.fn("ToolCallRepository.getCall.call")(function* (id) {
|
|
8731
|
+
var getCall = Effect37.fn("ToolCallRepository.getCall.call")(function* (executionId, id) {
|
|
8727
8732
|
const repository = yield* Service38;
|
|
8728
|
-
return yield* repository.getCall(id);
|
|
8733
|
+
return yield* repository.getCall(executionId, id);
|
|
8729
8734
|
});
|
|
8730
8735
|
var listCalls = Effect37.fn("ToolCallRepository.listCalls.call")(function* (executionId) {
|
|
8731
8736
|
const repository = yield* Service38;
|
|
@@ -8735,9 +8740,9 @@ var recordResult = Effect37.fn("ToolCallRepository.recordResult.call")(function*
|
|
|
8735
8740
|
const repository = yield* Service38;
|
|
8736
8741
|
return yield* repository.recordResult(input);
|
|
8737
8742
|
});
|
|
8738
|
-
var getResult = Effect37.fn("ToolCallRepository.getResult.call")(function* (callId) {
|
|
8743
|
+
var getResult = Effect37.fn("ToolCallRepository.getResult.call")(function* (executionId, callId) {
|
|
8739
8744
|
const repository = yield* Service38;
|
|
8740
|
-
return yield* repository.getResult(callId);
|
|
8745
|
+
return yield* repository.getResult(executionId, callId);
|
|
8741
8746
|
});
|
|
8742
8747
|
var ensureCall = Effect37.fn("ToolCallRepository.ensureCall.call")(function* (input) {
|
|
8743
8748
|
const repository = yield* Service38;
|
|
@@ -8767,9 +8772,9 @@ var acceptRemoteOutcome = Effect37.fn("ToolCallRepository.acceptRemoteOutcome.ca
|
|
|
8767
8772
|
const repository = yield* Service38;
|
|
8768
8773
|
return yield* repository.acceptRemoteOutcome(input);
|
|
8769
8774
|
});
|
|
8770
|
-
var listAttempts = Effect37.fn("ToolCallRepository.listAttempts.call")(function* (callId) {
|
|
8775
|
+
var listAttempts = Effect37.fn("ToolCallRepository.listAttempts.call")(function* (executionId, callId) {
|
|
8771
8776
|
const repository = yield* Service38;
|
|
8772
|
-
return yield* repository.listAttempts(callId);
|
|
8777
|
+
return yield* repository.listAttempts(executionId, callId);
|
|
8773
8778
|
});
|
|
8774
8779
|
var failOpenForExecution = Effect37.fn("ToolCallRepository.failOpenForExecution.call")(function* (input) {
|
|
8775
8780
|
const repository = yield* Service38;
|
|
@@ -10272,7 +10277,7 @@ var recordRequested = (repository, eventLog, input, tool) => repository.ensureCa
|
|
|
10272
10277
|
createdAt: input.createdAt
|
|
10273
10278
|
}).pipe(Effect43.mapError(mapRepositoryError3), Effect43.tap((call) => ensureToolEvent(eventLog, requestedEvent({ ...input, createdAt: call.createdAt }))));
|
|
10274
10279
|
var repairResultEvent = (eventLog, input, record2) => eventLog.maxSequence(input.executionId).pipe(Effect43.mapError(mapEventLogError2), Effect43.flatMap((max) => ensureToolEvent(eventLog, resultEvent({ ...input, createdAt: record2.createdAt - 1 }, record2.result, (max ?? input.eventSequence) + 1))));
|
|
10275
|
-
var recordReceived = (repository, eventLog, input, result) => repository.recordResult({ result, createdAt: input.createdAt + 1 }).pipe(Effect43.catchTag("DuplicateToolResult", () => repository.getResult(result.call_id).pipe(Effect43.mapError(mapRepositoryError3), Effect43.flatMap((existing) => existing === undefined ? Effect43.fail(ToolRuntimeError.make({ message: "DuplicateToolResult" })) : Effect43.succeed(existing)))), Effect43.mapError(preserveRecordResultError), Effect43.flatMap((record2) => eventLog.maxSequence(input.executionId).pipe(Effect43.mapError(mapEventLogError2), Effect43.flatMap((max) => appendIdempotentTo(eventLog, resultEvent(input, record2.result, (max ?? input.eventSequence) + 1)).pipe(Effect43.mapError(mapEventLogError2))), Effect43.as(record2.result))));
|
|
10280
|
+
var recordReceived = (repository, eventLog, input, result) => repository.recordResult({ executionId: input.executionId, result, createdAt: input.createdAt + 1 }).pipe(Effect43.catchTag("DuplicateToolResult", () => repository.getResult(input.executionId, result.call_id).pipe(Effect43.mapError(mapRepositoryError3), Effect43.flatMap((existing) => existing === undefined ? Effect43.fail(ToolRuntimeError.make({ message: "DuplicateToolResult" })) : Effect43.succeed(existing)))), Effect43.mapError(preserveRecordResultError), Effect43.flatMap((record2) => eventLog.maxSequence(input.executionId).pipe(Effect43.mapError(mapEventLogError2), Effect43.flatMap((max) => appendIdempotentTo(eventLog, resultEvent(input, record2.result, (max ?? input.eventSequence) + 1)).pipe(Effect43.mapError(mapEventLogError2))), Effect43.as(record2.result))));
|
|
10276
10281
|
var nonEmptyName = (name) => Schema40.decodeUnknownSync(exports_shared_schema.NonEmptyString)(name);
|
|
10277
10282
|
var needsApprovalFromEffectTool = (tool) => tool.needsApproval === undefined ? undefined : typeof tool.needsApproval === "boolean" ? tool.needsApproval : true;
|
|
10278
10283
|
var definitionFromEffectTool = (modelTool, options) => ({
|
|
@@ -10390,6 +10395,7 @@ var runExternalTool = (repository, waits, call, input) => Effect43.gen(function*
|
|
|
10390
10395
|
}).pipe(Effect43.mapError(mapWaitServiceError));
|
|
10391
10396
|
}
|
|
10392
10397
|
const waitingCall = call.state === "requested" ? yield* repository.prepareExternalWait({
|
|
10398
|
+
executionId: input.executionId,
|
|
10393
10399
|
callId: input.call.id,
|
|
10394
10400
|
waitId,
|
|
10395
10401
|
availableAt: input.createdAt,
|
|
@@ -10429,7 +10435,7 @@ var makeService = (initialTools) => Effect43.gen(function* () {
|
|
|
10429
10435
|
yield* ensureAllowed(registeredTool, input);
|
|
10430
10436
|
yield* validateInput(registeredTool, input);
|
|
10431
10437
|
const call = yield* recordRequested(repository, eventLog, input, registeredTool);
|
|
10432
|
-
const existingResult = yield* repository.getResult(input.call.id).pipe(Effect43.mapError(mapRepositoryError3));
|
|
10438
|
+
const existingResult = yield* repository.getResult(input.executionId, input.call.id).pipe(Effect43.mapError(mapRepositoryError3));
|
|
10433
10439
|
if (existingResult !== undefined) {
|
|
10434
10440
|
yield* repairResultEvent(eventLog, input, existingResult);
|
|
10435
10441
|
return existingResult.result;
|
|
@@ -10451,7 +10457,11 @@ var makeService = (initialTools) => Effect43.gen(function* () {
|
|
|
10451
10457
|
}
|
|
10452
10458
|
}
|
|
10453
10459
|
}
|
|
10454
|
-
const rawResult = call.placement.kind === "client" || call.placement.kind === "remote" ? yield* runExternalTool(repository, waits, call, effectiveInput) : yield* repository.beginAttempt({
|
|
10460
|
+
const rawResult = call.placement.kind === "client" || call.placement.kind === "remote" ? yield* runExternalTool(repository, waits, call, effectiveInput) : yield* repository.beginAttempt({
|
|
10461
|
+
executionId: input.executionId,
|
|
10462
|
+
callId: input.call.id,
|
|
10463
|
+
startedAt: effectiveInput.createdAt
|
|
10464
|
+
}).pipe(Effect43.mapError(mapRepositoryError3), Effect43.flatMap(() => finalToolResult(registeredTool, effectiveInput, call.idempotencyKey)), Effect43.catchTag("ToolExecutionFailed", (error5) => Effect43.succeed(placementFailureResult(input, error5.message))));
|
|
10455
10465
|
const result = input.transformResult === undefined || rawResult.error !== undefined ? rawResult : yield* input.transformResult(rawResult);
|
|
10456
10466
|
const received = yield* recordReceived(repository, eventLog, effectiveInput, result);
|
|
10457
10467
|
yield* recordToolCall(input.call.name, result.error === undefined ? "success" : "failure");
|
|
@@ -15467,11 +15477,24 @@ var reconcileCompactionCheckpoint = Effect81.fn("AgentLoop.reconcileCompactionCh
|
|
|
15467
15477
|
const checkpoint = path2.at(-1);
|
|
15468
15478
|
if (checkpoint?._tag !== "Compaction" || checkpoint.version !== 2)
|
|
15469
15479
|
return restoredHistory;
|
|
15480
|
+
const entryScope = input.sessionEntryScope ?? input.executionId;
|
|
15481
|
+
if (!checkpoint.id.startsWith(`${input.sessionId}:entry:${entryScope}:`))
|
|
15482
|
+
return restoredHistory;
|
|
15483
|
+
const result = {
|
|
15484
|
+
_tag: "AlreadyPresent",
|
|
15485
|
+
checkpoint,
|
|
15486
|
+
leafId: checkpoint.id
|
|
15487
|
+
};
|
|
15488
|
+
const repaired = exports_session.buildContext(path2);
|
|
15489
|
+
if (restoredHistory === undefined) {
|
|
15490
|
+
yield* persistTranscript(chats, input, repaired);
|
|
15491
|
+
yield* onCheckpointCommitted(result);
|
|
15492
|
+
return repaired;
|
|
15493
|
+
}
|
|
15470
15494
|
const before = exports_session.buildContext(path2.slice(0, -1));
|
|
15471
15495
|
if (!promptEquivalence2(before, restoredHistory)) {
|
|
15472
|
-
|
|
15473
|
-
|
|
15474
|
-
yield* onCheckpointCommitted(checkpoint);
|
|
15496
|
+
if (promptEquivalence2(repaired, restoredHistory)) {
|
|
15497
|
+
yield* onCheckpointCommitted(result);
|
|
15475
15498
|
return restoredHistory;
|
|
15476
15499
|
}
|
|
15477
15500
|
if (!failOnMismatch)
|
|
@@ -15481,9 +15504,8 @@ var reconcileCompactionCheckpoint = Effect81.fn("AgentLoop.reconcileCompactionCh
|
|
|
15481
15504
|
next_event_sequence: input.eventSequence + 1
|
|
15482
15505
|
});
|
|
15483
15506
|
}
|
|
15484
|
-
const repaired = exports_session.buildContext(path2);
|
|
15485
15507
|
yield* persistTranscript(chats, input, repaired);
|
|
15486
|
-
yield* onCheckpointCommitted(
|
|
15508
|
+
yield* onCheckpointCommitted(result);
|
|
15487
15509
|
return repaired;
|
|
15488
15510
|
});
|
|
15489
15511
|
var toolCallDecodeMaxRetriesConfig = Config4.int("RELAY_AGENT_TOOL_CALL_DECODE_MAX_RETRIES").pipe(Config4.withDefault(2));
|
|
@@ -15631,7 +15653,7 @@ var compactionOptions = (agent) => agent.compaction_policy === undefined ? undef
|
|
|
15631
15653
|
reserveTokens: agent.compaction_policy.reserve_tokens,
|
|
15632
15654
|
keepRecentTokens: agent.compaction_policy.keep_recent_tokens
|
|
15633
15655
|
};
|
|
15634
|
-
var
|
|
15656
|
+
var make11 = (layerOptions) => Effect81.gen(function* () {
|
|
15635
15657
|
const eventLog = yield* Service41;
|
|
15636
15658
|
const languageModels = yield* Service8;
|
|
15637
15659
|
const tools = yield* Service12;
|
|
@@ -15714,12 +15736,12 @@ var layer58 = Layer71.effect(Service61, Effect81.gen(function* () {
|
|
|
15714
15736
|
input: jsonValue6(resumeSuspension.tool_params)
|
|
15715
15737
|
};
|
|
15716
15738
|
const storedHistory = resumeToolCall === undefined ? yield* loadStoredHistory(chats, input) : yield* restoreHistory(chats, input);
|
|
15717
|
-
const
|
|
15718
|
-
const durableMaxSequence = continuingDurableHistory ? yield* eventLog.maxSequence(input.executionId).pipe(Effect81.mapError((error5) => loopError(error5, input.eventSequence + 1))) : undefined;
|
|
15739
|
+
const durableMaxSequence = yield* eventLog.maxSequence(input.executionId).pipe(Effect81.mapError((error5) => loopError(error5, input.eventSequence + 1)));
|
|
15719
15740
|
const nextEventSequence5 = Math.max(input.eventSequence + 1, (durableMaxSequence ?? -1) + 1);
|
|
15720
15741
|
const allocator = yield* make10(nextEventSequence5);
|
|
15721
|
-
const onCheckpointCommitted = (
|
|
15722
|
-
const durableHistory =
|
|
15742
|
+
const onCheckpointCommitted = (result) => appendCompactionCommittedEvent(eventLog, allocator, input, result.checkpoint).pipe(Effect81.andThen(layerOptions?.onCheckpointCommitted?.(result) ?? Effect81.void));
|
|
15743
|
+
const durableHistory = Option19.isNone(sessionRepository) ? storedHistory : yield* reconcileCompactionCheckpoint(sessionRepository.value, chats, input, storedHistory, onCheckpointCommitted, resumeSuspension === undefined);
|
|
15744
|
+
const continuingDurableHistory = durableHistory !== undefined;
|
|
15723
15745
|
const permissionsLayer = input.agent.permission_rules === undefined ? Layer71.empty : Option19.isNone(permissionRules) || Option19.isNone(waits) ? yield* AgentLoopError.make({
|
|
15724
15746
|
message: "Execution permission rules require PermissionRuleRepository and WaitService in context",
|
|
15725
15747
|
next_event_sequence: input.eventSequence + 1
|
|
@@ -15749,6 +15771,7 @@ var layer58 = Layer71.effect(Service61, Effect81.gen(function* () {
|
|
|
15749
15771
|
const seededTokens = continuingDurableHistory ? yield* eventLog.sumUsage(input.executionId).pipe(Effect81.mapError((error5) => loopError(error5, input.eventSequence + 1))) : 0;
|
|
15750
15772
|
const memory = input.memorySubjectId === undefined ? undefined : { key: { agent: input.agent.name, subject: input.memorySubjectId } };
|
|
15751
15773
|
const restoredEpoch = durableHistory !== undefined && Option19.isSome(contextEpochs) ? yield* restoreContextEpoch(contextEpochs.value, input) : undefined;
|
|
15774
|
+
const freshSessionHistory = durableHistory !== undefined || input.sessionId === undefined || Option19.isNone(sessionRepository) ? undefined : yield* sessionRepository.value.path({ sessionId: input.sessionId }).pipe(Effect81.mapError((error5) => loopError(error5, input.eventSequence + 1)), Effect81.map((path2) => path2.length === 0 ? undefined : exports_session.buildContext(path2)));
|
|
15752
15775
|
const { options, instructionsLayer } = durableHistory === undefined ? yield* assembler.assemble({ agent: input.agent, tools: definitions2, input: input.input }).pipe(Effect81.mapError((error5) => loopError(error5, input.eventSequence + 1)), Effect81.flatMap((assembled) => Effect81.gen(function* () {
|
|
15753
15776
|
const epoch = { baseline: assembled.system, dynamicSourceIds: [] };
|
|
15754
15777
|
if (Option19.isSome(contextEpochs))
|
|
@@ -15756,10 +15779,11 @@ var layer58 = Layer71.effect(Service61, Effect81.gen(function* () {
|
|
|
15756
15779
|
return {
|
|
15757
15780
|
options: {
|
|
15758
15781
|
prompt: assembled.prompt,
|
|
15782
|
+
...freshSessionHistory === undefined ? {} : { history: freshSessionHistory },
|
|
15759
15783
|
sessionId: input.sessionId ?? input.executionId,
|
|
15760
15784
|
...memory === undefined ? {} : { memory }
|
|
15761
15785
|
},
|
|
15762
|
-
instructionsLayer: layerFromEpoch(epoch)
|
|
15786
|
+
instructionsLayer: freshSessionHistory === undefined ? layerFromEpoch(epoch) : Layer71.empty
|
|
15763
15787
|
};
|
|
15764
15788
|
}))) : {
|
|
15765
15789
|
options: {
|
|
@@ -15910,7 +15934,7 @@ var layer58 = Layer71.effect(Service61, Effect81.gen(function* () {
|
|
|
15910
15934
|
const runFold = Effect81.scoped(Effect81.gen(function* () {
|
|
15911
15935
|
const base = withToolCallDecodeResilience(yield* LanguageModel.LanguageModel, definitions2.map((definition) => definition.name), toolCallDecodeMaxRetries, (classification) => Ref18.set(lastModelFailureClassification, classification));
|
|
15912
15936
|
const sessionStore = input.sessionId !== undefined && Option19.isSome(sessionRepository) ? yield* make4(input.sessionId, input.sessionEntryScope ?? input.executionId, {
|
|
15913
|
-
onCheckpointCommitted: (result) => onCheckpointCommitted(result
|
|
15937
|
+
onCheckpointCommitted: (result) => onCheckpointCommitted(result).pipe(Effect81.orDie)
|
|
15914
15938
|
}).pipe(Effect81.provideService(exports_session_repository.Service, sessionRepository.value)) : undefined;
|
|
15915
15939
|
const foldContext = yield* Layer71.build(Layer71.mergeAll(executorLayer, handlerLayer, layer53, exports_model_middleware.identityLayer, instructionsLayer, permissionsLayer, steeringLayer));
|
|
15916
15940
|
const folded = exports_agent.stream(agent, batonOptions).pipe(Stream9.runFoldEffect(() => ({
|
|
@@ -15952,7 +15976,9 @@ var layer58 = Layer71.effect(Service61, Effect81.gen(function* () {
|
|
|
15952
15976
|
};
|
|
15953
15977
|
})
|
|
15954
15978
|
});
|
|
15955
|
-
})
|
|
15979
|
+
});
|
|
15980
|
+
var layerWith = (layerOptions) => Layer71.effect(Service61, make11(layerOptions));
|
|
15981
|
+
var layer58 = layerWith();
|
|
15956
15982
|
var run5 = Effect81.fn("AgentLoop.run.call")(function* (input) {
|
|
15957
15983
|
const service = yield* Service61;
|
|
15958
15984
|
return yield* service.run(input);
|
|
@@ -16285,7 +16311,7 @@ var nextEventSequence5 = Effect86.fn("SchedulerService.nextEventSequence")(funct
|
|
|
16285
16311
|
const max = yield* eventLog.maxSequence(executionId).pipe(Effect86.mapError(toSchedulerError));
|
|
16286
16312
|
return max === undefined ? 0 : max + 1;
|
|
16287
16313
|
});
|
|
16288
|
-
var
|
|
16314
|
+
var make12 = Effect86.gen(function* () {
|
|
16289
16315
|
const schedules = yield* exports_schedule_repository.Service;
|
|
16290
16316
|
const waits = yield* Service42;
|
|
16291
16317
|
const eventLog = yield* Service41;
|
|
@@ -16393,7 +16419,7 @@ var pollLoop = (service, pollIntervalMillis) => Effect86.gen(function* () {
|
|
|
16393
16419
|
}
|
|
16394
16420
|
});
|
|
16395
16421
|
var layer62 = Layer76.effect(Service69, Effect86.gen(function* () {
|
|
16396
|
-
const service = yield*
|
|
16422
|
+
const service = yield* make12;
|
|
16397
16423
|
const enabled3 = yield* enabledConfig;
|
|
16398
16424
|
const pollIntervalMillis = yield* pollIntervalMillisConfig;
|
|
16399
16425
|
if (enabled3) {
|
|
@@ -16401,7 +16427,7 @@ var layer62 = Layer76.effect(Service69, Effect86.gen(function* () {
|
|
|
16401
16427
|
}
|
|
16402
16428
|
return service;
|
|
16403
16429
|
}));
|
|
16404
|
-
var memoryLayer45 = Layer76.effect(Service69,
|
|
16430
|
+
var memoryLayer45 = Layer76.effect(Service69, make12);
|
|
16405
16431
|
var runOnce2 = Effect86.fn("SchedulerService.runOnce.call")(function* () {
|
|
16406
16432
|
const service = yield* Service69;
|
|
16407
16433
|
return yield* service.runOnce;
|
|
@@ -16556,8 +16582,8 @@ var parentNotifierLayer = Layer77.effect(Service54, Effect87.gen(function* () {
|
|
|
16556
16582
|
}).pipe(Effect87.provideService(exports_execution_repository.Service, executionRepository))
|
|
16557
16583
|
});
|
|
16558
16584
|
}));
|
|
16559
|
-
var agentLoopLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer) =>
|
|
16560
|
-
var runtimeServicesLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer) => {
|
|
16585
|
+
var agentLoopLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, options) => layerWith(options).pipe(Layer77.provideMerge(toolRuntimeLayer), Layer77.provideMerge(layer40), Layer77.provideMerge(layer42()), Layer77.provideMerge(promptAssemblerLayer), Layer77.provideMerge(schemaRegistryLayer));
|
|
16586
|
+
var runtimeServicesLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer, agentLoopOptions) => {
|
|
16561
16587
|
const inboxLayer = Layer77.unwrap(Effect87.gen(function* () {
|
|
16562
16588
|
const makeExecutionClient = yield* client;
|
|
16563
16589
|
const executionRepository = yield* exports_execution_repository.Service;
|
|
@@ -16565,13 +16591,13 @@ var runtimeServicesLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRe
|
|
|
16565
16591
|
}));
|
|
16566
16592
|
const envelopeLayer = layer52.pipe(Layer77.provideMerge(layerFromRepository3), Layer77.provideMerge(inboxLayer));
|
|
16567
16593
|
const topicLayer = layer51.pipe(Layer77.provideMerge(inboxLayer));
|
|
16568
|
-
const agentLayer = agentLoopLayerWith(toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer).pipe(Layer77.provideMerge(inboxLayer), Layer77.provideMerge(envelopeLayer), Layer77.provideMerge(topicLayer));
|
|
16594
|
+
const agentLayer = agentLoopLayerWith(toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, agentLoopOptions).pipe(Layer77.provideMerge(inboxLayer), Layer77.provideMerge(envelopeLayer), Layer77.provideMerge(topicLayer));
|
|
16569
16595
|
const entityRegistryLayer = layer60.pipe(Layer77.provideMerge(layer39));
|
|
16570
16596
|
const entityInstanceLayer = layer61.pipe(Layer77.provideMerge(entityRegistryLayer), Layer77.provideMerge(layerFromRepository3), Layer77.provideMerge(layerFromRepository));
|
|
16571
16597
|
return Layer77.mergeAll(addressResolutionLayerWith(toolRuntimeLayer), layer45, executionServiceLayer, layerFromServices, layerFromServices2, layerFromRepository2, agentLayer, layer48, parentNotifierLayer, inboxLayer, envelopeLayer, topicLayer, entityRegistryLayer, toolTransitionCoordinatorLayer, layer47).pipe(Layer77.provideMerge(entityInstanceLayer), Layer77.provideMerge(layer38), Layer77.provideMerge(layer34), Layer77.provideMerge(layerFromRepository));
|
|
16572
16598
|
};
|
|
16573
|
-
var workflowLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer) => layer49.pipe(Layer77.provideMerge(runtimeServicesLayerWith(toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer)));
|
|
16574
|
-
var workflowAndEntityLayerWith = (toolRuntimeLayer, schedulerLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer) => Layer77.mergeAll(layer59, schedulerLayer).pipe(Layer77.provideMerge(workflowLayerWith(toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer)));
|
|
16599
|
+
var workflowLayerWith = (toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer, agentLoopOptions) => layer49.pipe(Layer77.provideMerge(runtimeServicesLayerWith(toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer, agentLoopOptions)));
|
|
16600
|
+
var workflowAndEntityLayerWith = (toolRuntimeLayer, schedulerLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer, agentLoopOptions) => Layer77.mergeAll(layer59, schedulerLayer).pipe(Layer77.provideMerge(workflowLayerWith(toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, toolTransitionCoordinatorLayer, agentLoopOptions)));
|
|
16575
16601
|
var makeService2 = (database, checkDatabase) => Effect87.gen(function* () {
|
|
16576
16602
|
const sharding = yield* Sharding.Sharding;
|
|
16577
16603
|
const workflow = yield* WorkflowEngine2.WorkflowEngine;
|
|
@@ -16660,7 +16686,7 @@ var testLanguageModelLayer = layerFromRegistrationEffects2([
|
|
|
16660
16686
|
]).pipe(Layer77.orDie);
|
|
16661
16687
|
var testEmbeddingModelLayer = deterministicTestLayer();
|
|
16662
16688
|
var defaultPromptAssemblerLayer = (blobStoreLayer = passthroughLayer2, artifactStoreLayer = passthroughLayer) => defaultLayerWithStores.pipe(Layer77.provide(Layer77.mergeAll(blobStoreLayer, artifactStoreLayer)));
|
|
16663
|
-
var
|
|
16689
|
+
var layerWith2 = (options) => {
|
|
16664
16690
|
const blobStoreLayer = options.blobStoreLayer ?? passthroughLayer2;
|
|
16665
16691
|
const artifactStoreLayer = options.artifactStoreLayer ?? passthroughLayer;
|
|
16666
16692
|
const promptAssemblerLayer = options.promptAssemblerLayer ?? defaultPromptAssemblerLayer(blobStoreLayer, artifactStoreLayer);
|
|
@@ -16669,7 +16695,7 @@ var layerWith = (options) => {
|
|
|
16669
16695
|
const memoryLayer46 = layer41().pipe(Layer77.provideMerge(embeddingModelLayer));
|
|
16670
16696
|
const clusterContextLayer = Layer77.mergeAll(options.languageModelLayer, memoryLayer46, blobStoreLayer, artifactStoreLayer, promptAssemblerLayer, schemaRegistryLayer);
|
|
16671
16697
|
const clusterLayer = options.clusterLayer.pipe(Layer77.provide(clusterContextLayer));
|
|
16672
|
-
return Layer77.mergeAll(options.checkLayer, workflowAndEntityLayerWith(options.toolRuntimeLayer, options.schedulerLayer, promptAssemblerLayer, schemaRegistryLayer, options.toolTransitionCoordinatorLayer)).pipe(Layer77.provideMerge(options.languageModelLayer), Layer77.provide(memoryLayer46), Layer77.provideMerge(ClusterWorkflowEngine.layer), Layer77.provideMerge(clusterLayer), Layer77.provideMerge(options.repositoryLayer), Layer77.provideMerge(blobStoreLayer), Layer77.provideMerge(artifactStoreLayer));
|
|
16698
|
+
return Layer77.mergeAll(options.checkLayer, workflowAndEntityLayerWith(options.toolRuntimeLayer, options.schedulerLayer, promptAssemblerLayer, schemaRegistryLayer, options.toolTransitionCoordinatorLayer, options.onCheckpointCommitted === undefined ? undefined : { onCheckpointCommitted: options.onCheckpointCommitted })).pipe(Layer77.provideMerge(options.languageModelLayer), Layer77.provide(memoryLayer46), Layer77.provideMerge(ClusterWorkflowEngine.layer), Layer77.provideMerge(clusterLayer), Layer77.provideMerge(options.repositoryLayer), Layer77.provideMerge(blobStoreLayer), Layer77.provideMerge(artifactStoreLayer));
|
|
16673
16699
|
};
|
|
16674
16700
|
var layerWithClient = (options) => {
|
|
16675
16701
|
const blobStoreLayer = options.blobStoreLayer ?? passthroughLayer2;
|
|
@@ -16681,7 +16707,7 @@ var layerWithClient = (options) => {
|
|
|
16681
16707
|
return Layer77.mergeAll(options.checkLayer, runtimeServicesLayerWith(options.toolRuntimeLayer, promptAssemblerLayer, schemaRegistryLayer, options.toolTransitionCoordinatorLayer)).pipe(Layer77.provideMerge(options.languageModelLayer), Layer77.provide(memoryLayer46), Layer77.provideMerge(options.clusterLayer), Layer77.provideMerge(options.repositoryLayer), Layer77.provideMerge(blobStoreLayer), Layer77.provideMerge(artifactStoreLayer));
|
|
16682
16708
|
};
|
|
16683
16709
|
var layerWithServices = (options) => {
|
|
16684
|
-
const runtime =
|
|
16710
|
+
const runtime = layerWith2({
|
|
16685
16711
|
toolTransitionCoordinatorLayer: sqlLayer,
|
|
16686
16712
|
checkLayer: sqlCheckLayer,
|
|
16687
16713
|
clusterLayer: sqlClusterLayer,
|
|
@@ -16693,12 +16719,13 @@ var layerWithServices = (options) => {
|
|
|
16693
16719
|
promptAssemblerLayer: options.promptAssemblerLayer,
|
|
16694
16720
|
blobStoreLayer: options.blobStoreLayer,
|
|
16695
16721
|
artifactStoreLayer: options.artifactStoreLayer,
|
|
16696
|
-
schemaRegistryLayer: options.schemaRegistryLayer
|
|
16722
|
+
schemaRegistryLayer: options.schemaRegistryLayer,
|
|
16723
|
+
...options.onCheckpointCommitted === undefined ? {} : { onCheckpointCommitted: options.onCheckpointCommitted }
|
|
16697
16724
|
});
|
|
16698
16725
|
return runtime.pipe(Layer77.provide(options.databaseLayer));
|
|
16699
16726
|
};
|
|
16700
16727
|
var layerWithServicesMultiNode = (options) => {
|
|
16701
|
-
const runtime =
|
|
16728
|
+
const runtime = layerWith2({
|
|
16702
16729
|
toolTransitionCoordinatorLayer: sqlLayer,
|
|
16703
16730
|
checkLayer: Layer77.mergeAll(sqlCheckLayer, multiNodeDialectGuard),
|
|
16704
16731
|
clusterLayer: clusterLayerHttp(options.cluster),
|
|
@@ -16710,7 +16737,8 @@ var layerWithServicesMultiNode = (options) => {
|
|
|
16710
16737
|
promptAssemblerLayer: options.promptAssemblerLayer,
|
|
16711
16738
|
blobStoreLayer: options.blobStoreLayer,
|
|
16712
16739
|
artifactStoreLayer: options.artifactStoreLayer,
|
|
16713
|
-
schemaRegistryLayer: options.schemaRegistryLayer
|
|
16740
|
+
schemaRegistryLayer: options.schemaRegistryLayer,
|
|
16741
|
+
...options.onCheckpointCommitted === undefined ? {} : { onCheckpointCommitted: options.onCheckpointCommitted }
|
|
16714
16742
|
});
|
|
16715
16743
|
return runtime.pipe(Layer77.provide(options.databaseLayer));
|
|
16716
16744
|
};
|
|
@@ -16730,7 +16758,7 @@ var layerWithServicesMultiNodeClientOnly = (options) => {
|
|
|
16730
16758
|
});
|
|
16731
16759
|
return runtime.pipe(Layer77.provide(options.databaseLayer));
|
|
16732
16760
|
};
|
|
16733
|
-
var layer63 =
|
|
16761
|
+
var layer63 = layerWith2({
|
|
16734
16762
|
toolTransitionCoordinatorLayer: sqlLayer,
|
|
16735
16763
|
checkLayer: sqlCheckLayer,
|
|
16736
16764
|
clusterLayer: sqlClusterLayer,
|
|
@@ -16740,7 +16768,7 @@ var layer63 = layerWith({
|
|
|
16740
16768
|
toolRuntimeLayer: layer35(),
|
|
16741
16769
|
schedulerLayer: layer62
|
|
16742
16770
|
});
|
|
16743
|
-
var testLayerWithServices = (options) =>
|
|
16771
|
+
var testLayerWithServices = (options) => layerWith2({
|
|
16744
16772
|
toolTransitionCoordinatorLayer: memoryLayer41,
|
|
16745
16773
|
checkLayer: memoryCheckLayer,
|
|
16746
16774
|
clusterLayer: memoryClusterLayer,
|
|
@@ -17923,7 +17951,7 @@ var layerFromRuntime3 = Layer80.effect(Service, Effect90.gen(function* () {
|
|
|
17923
17951
|
return yield* ExecutionNotFound.make({ execution_id: executionId });
|
|
17924
17952
|
const openWaits = yield* envelopeReady.listAllWaits({ executionId, state: "open" }).pipe(Effect90.mapError(toClientError));
|
|
17925
17953
|
const calls = yield* toolCalls2.listCalls(executionId).pipe(Effect90.mapError(toClientError));
|
|
17926
|
-
const pendingChunks = yield* Effect90.forEach(calls, (call) => toolCalls2.getResult(call.call.id).pipe(Effect90.mapError(toClientError), Effect90.map((result) => result === undefined ? [toToolCallSummary(call)] : [])));
|
|
17954
|
+
const pendingChunks = yield* Effect90.forEach(calls, (call) => toolCalls2.getResult(call.executionId, call.call.id).pipe(Effect90.mapError(toClientError), Effect90.map((result) => result === undefined ? [toToolCallSummary(call)] : [])));
|
|
17927
17955
|
const children = yield* childExecutions.listByExecution(executionId).pipe(Effect90.mapError(toClientError));
|
|
17928
17956
|
const maxSequence3 = yield* eventLog.maxSequence(executionId).pipe(Effect90.mapError(toClientError));
|
|
17929
17957
|
const lastEvent = maxSequence3 === undefined ? undefined : yield* eventLog.findBySequence({ executionId, sequence: maxSequence3 }).pipe(Effect90.mapError(toClientError), Effect90.map(Option23.getOrUndefined));
|
|
@@ -18389,13 +18417,13 @@ var layerFromRuntime3 = Layer80.effect(Service, Effect90.gen(function* () {
|
|
|
18389
18417
|
state: "open",
|
|
18390
18418
|
...input.execution_id === undefined ? {} : { executionId: input.execution_id }
|
|
18391
18419
|
}).pipe(Effect90.mapError(toClientError));
|
|
18392
|
-
const records = yield* Effect90.forEach(openWaits.filter((wait) => wait.metadata.kind === "tool-placement"), (wait) => toolCalls2.getCall(exports_ids_schema.ToolCallId.make(String(wait.metadata.tool_call_id ?? ""))).pipe(Effect90.mapError(toClientError), Effect90.map((call) => call?.placement.kind === "client" && call.state === "waiting" ? toPendingToolCall(call) : undefined)));
|
|
18420
|
+
const records = yield* Effect90.forEach(openWaits.filter((wait) => wait.metadata.kind === "tool-placement"), (wait) => toolCalls2.getCall(wait.executionId, exports_ids_schema.ToolCallId.make(String(wait.metadata.tool_call_id ?? ""))).pipe(Effect90.mapError(toClientError), Effect90.map((call) => call?.placement.kind === "client" && call.state === "waiting" ? toPendingToolCall(call) : undefined)));
|
|
18393
18421
|
const sorted = records.filter((record2) => record2 !== undefined).toSorted((left, right) => left.requested_at - right.requested_at || left.call.id.localeCompare(right.call.id));
|
|
18394
18422
|
return { tool_calls: input.limit === undefined ? sorted : sorted.slice(0, input.limit) };
|
|
18395
18423
|
}),
|
|
18396
18424
|
fulfillToolCall: Effect90.fn("Client.runtime.fulfillToolCall")(function* (input) {
|
|
18397
18425
|
const coordinated = yield* coordinateToolTransition(Effect90.gen(function* () {
|
|
18398
|
-
const call = yield* toolCalls2.getCall(input.tool_call_id).pipe(Effect90.mapError(toClientError));
|
|
18426
|
+
const call = yield* toolCalls2.getCall(input.execution_id, input.tool_call_id).pipe(Effect90.mapError(toClientError));
|
|
18399
18427
|
if (call === undefined || call.executionId !== input.execution_id || call.placement.kind !== "client") {
|
|
18400
18428
|
return yield* ClientError.make({ message: `Client tool call not found: ${input.tool_call_id}` });
|
|
18401
18429
|
}
|
|
@@ -18422,7 +18450,7 @@ var layerFromRuntime3 = Layer80.effect(Service, Effect90.gen(function* () {
|
|
|
18422
18450
|
}),
|
|
18423
18451
|
completeToolWork: Effect90.fn("Client.runtime.completeToolWork")(function* (input) {
|
|
18424
18452
|
const coordinated = yield* coordinateToolTransition(Effect90.gen(function* () {
|
|
18425
|
-
const call = yield* toolCalls2.getCall(input.tool_call_id).pipe(Effect90.mapError(toClientError));
|
|
18453
|
+
const call = yield* toolCalls2.getCall(input.execution_id, input.tool_call_id).pipe(Effect90.mapError(toClientError));
|
|
18426
18454
|
const active = call !== undefined && call.placement.kind === "remote" && call.state === "running" && call.activeAttemptId === input.attempt_id && call.claimOwner === input.worker_id && call.claimExpiresAt !== undefined && call.claimExpiresAt > input.completed_at;
|
|
18427
18455
|
const acceptedRetry = call !== undefined && call.placement.kind === "remote" && call.state === "submitted" && call.activeAttemptId === input.attempt_id && call.externalOutcome !== undefined;
|
|
18428
18456
|
if (call === undefined || !active && !acceptedRetry) {
|
|
@@ -18430,6 +18458,7 @@ var layerFromRuntime3 = Layer80.effect(Service, Effect90.gen(function* () {
|
|
|
18430
18458
|
}
|
|
18431
18459
|
const wait = yield* resolveToolOutcomeWait(waits, eventLog, call, input.outcome, input.completed_at);
|
|
18432
18460
|
const accepted2 = yield* toolCalls2.acceptRemoteOutcome({
|
|
18461
|
+
executionId: input.execution_id,
|
|
18433
18462
|
callId: input.tool_call_id,
|
|
18434
18463
|
attemptId: input.attempt_id,
|
|
18435
18464
|
workerId: input.worker_id,
|
|
@@ -18443,6 +18472,7 @@ var layerFromRuntime3 = Layer80.effect(Service, Effect90.gen(function* () {
|
|
|
18443
18472
|
}),
|
|
18444
18473
|
releaseToolWork: Effect90.fn("Client.runtime.releaseToolWork")(function* (input) {
|
|
18445
18474
|
const call = yield* coordinateToolTransition(toolCalls2.releaseRemote({
|
|
18475
|
+
executionId: input.execution_id,
|
|
18446
18476
|
callId: input.tool_call_id,
|
|
18447
18477
|
attemptId: input.attempt_id,
|
|
18448
18478
|
workerId: input.worker_id,
|
|
@@ -18457,7 +18487,7 @@ var layerFromRuntime3 = Layer80.effect(Service, Effect90.gen(function* () {
|
|
|
18457
18487
|
};
|
|
18458
18488
|
}),
|
|
18459
18489
|
listToolAttempts: Effect90.fn("Client.runtime.listToolAttempts")(function* (input) {
|
|
18460
|
-
const attempts = yield* toolCalls2.listAttempts(input.tool_call_id).pipe(Effect90.mapError(toClientError));
|
|
18490
|
+
const attempts = yield* toolCalls2.listAttempts(input.execution_id, input.tool_call_id).pipe(Effect90.mapError(toClientError));
|
|
18461
18491
|
return { attempts: attempts.map(toToolAttempt) };
|
|
18462
18492
|
}),
|
|
18463
18493
|
submitInboundEnvelope: Effect90.fn("Client.runtime.submitInboundEnvelope")(function* (input) {
|
|
@@ -18574,7 +18604,7 @@ class Service73 extends Context73.Service()("@relayfx/sdk/runtime-database/Servi
|
|
|
18574
18604
|
// src/runtime-database-owner.ts
|
|
18575
18605
|
import { Effect as Effect91, Layer as Layer81 } from "effect";
|
|
18576
18606
|
var internals = new WeakMap;
|
|
18577
|
-
var
|
|
18607
|
+
var make13 = (value) => {
|
|
18578
18608
|
const database = {
|
|
18579
18609
|
dialect: value.dialect,
|
|
18580
18610
|
migrate: value.migrate,
|
|
@@ -18846,7 +18876,8 @@ function layerEmbedded(options) {
|
|
|
18846
18876
|
blobStoreLayer: options.blobStoreLayer,
|
|
18847
18877
|
artifactStoreLayer: options.artifactStoreLayer,
|
|
18848
18878
|
promptAssemblerLayer: options.promptAssemblerLayer,
|
|
18849
|
-
schemaRegistryLayer: options.schemaRegistryLayer
|
|
18879
|
+
schemaRegistryLayer: options.schemaRegistryLayer,
|
|
18880
|
+
...options.onCheckpointCommitted === undefined ? {} : { onCheckpointCommitted: options.onCheckpointCommitted }
|
|
18850
18881
|
});
|
|
18851
18882
|
const host = normalizeHost(publicGraph(withHosts(runner, options), runtimeLayer(options.database), "embedded", {
|
|
18852
18883
|
fanOut: options.childFanOutHostLayer !== undefined,
|
|
@@ -18864,6 +18895,7 @@ var layerRunner = (options) => {
|
|
|
18864
18895
|
artifactStoreLayer: options.artifactStoreLayer,
|
|
18865
18896
|
promptAssemblerLayer: options.promptAssemblerLayer,
|
|
18866
18897
|
schemaRegistryLayer: options.schemaRegistryLayer,
|
|
18898
|
+
...options.onCheckpointCommitted === undefined ? {} : { onCheckpointCommitted: options.onCheckpointCommitted },
|
|
18867
18899
|
cluster: options.cluster
|
|
18868
18900
|
});
|
|
18869
18901
|
return normalizeHost(publicGraph(runner, runtimeLayer(options.database), "runner"), "runner", options.database.dialect);
|
|
@@ -18879,4 +18911,4 @@ var layerClient = (options) => {
|
|
|
18879
18911
|
return embeddedOutput(normalizeHost(publicGraph(runner, runtimeLayer(options.database), "client"), "client", options.database.dialect));
|
|
18880
18912
|
};
|
|
18881
18913
|
|
|
18882
|
-
export { exports_child_fan_out_host, exports_artifact_store_service, exports_blob_store_service, exports_prompt_assembler_service, Dialect, dialect, fromDbTimestamp, fromNullableDbTimestamp, timestampParam, encodeJson, decodeJson, decodeBool, exports_embedding_model_service, exports_language_model_service, exports_model_hub, exports_schema_registry_service, RuntimeConfigurationError, normalizeAcquisitionCause, DatabaseIdentity, makeDatabaseIdentity, exports_workflow_definition_host, Service73 as Service,
|
|
18914
|
+
export { exports_child_fan_out_host, exports_artifact_store_service, exports_blob_store_service, exports_prompt_assembler_service, Dialect, dialect, fromDbTimestamp, fromNullableDbTimestamp, timestampParam, encodeJson, decodeJson, decodeBool, exports_embedding_model_service, exports_language_model_service, exports_model_hub, exports_schema_registry_service, RuntimeConfigurationError, normalizeAcquisitionCause, DatabaseIdentity, makeDatabaseIdentity, exports_workflow_definition_host, Service73 as Service, make13 as make, runtimeLayer, exports_tool_runtime, SqlFailure, DatabaseAlreadyOwned, MigratorError, SchemaHeadMismatch, RuntimeTopologyError, RuntimeMigrationError, exports_runtime };
|