Spaces:
Sleeping
Sleeping
File size: 3,688 Bytes
90cbf22 df2ef4f 90cbf22 df2ef4f 90cbf22 df2ef4f 90cbf22 c19005f 26088c6 90cbf22 df2ef4f 90cbf22 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
import { v } from 'convex/values';
import { internal } from './_generated/api';
import { DatabaseReader, MutationCtx, mutation } from './_generated/server';
import { Descriptions } from '../data/characters';
import * as map from '../data/gentle';
import { insertInput } from './aiTown/insertInput';
import { Id } from './_generated/dataModel';
import { createEngine } from './aiTown/main';
import { ENGINE_ACTION_DURATION, MAX_NPC } from './constants';
import { assertApiKey } from './util/llm';
const init = mutation({
args: {
numAgents: v.optional(v.number()),
},
handler: async (ctx, args) => {
assertApiKey();
const { worldStatus, engine } = await getOrCreateDefaultWorld(ctx);
if (worldStatus.status !== 'running') {
console.warn(
`Engine ${engine._id} is not active! Run "npx convex run testing:resume" to restart it.`,
);
return;
}
const shouldCreate = await shouldCreateAgents(
ctx.db,
worldStatus.worldId,
worldStatus.engineId,
);
if (shouldCreate) {
const toCreate = args.numAgents !== undefined ? args.numAgents : MAX_NPC; //Descriptions.length;
for (let i = 0; i < toCreate; i++) {
await insertInput(ctx, worldStatus.worldId, 'createAgent', {
descriptionIndex: i % Descriptions.length,
type: 'villager',
});
}
}
},
});
export default init;
async function getOrCreateDefaultWorld(ctx: MutationCtx) {
const now = Date.now();
let worldStatus = await ctx.db
.query('worldStatus')
.filter((q) => q.eq(q.field('isDefault'), true))
.unique();
if (worldStatus) {
const engine = (await ctx.db.get(worldStatus.engineId))!;
return { worldStatus, engine };
}
const engineId = await createEngine(ctx);
const engine = (await ctx.db.get(engineId))!;
const worldId = await ctx.db.insert('worlds', {
nextId: 0,
agents: [],
conversations: [],
players: [],
playersInit: [],
// initialize game cycle counter
gameCycle: {
currentTime: 0,
cycleState: 'LobbyState',
cycleIndex: -1,
cycleNumber:0,
},
gameVotes: [],
llmVotes: []
});
const worldStatusId = await ctx.db.insert('worldStatus', {
engineId: engineId,
isDefault: true,
lastViewed: now,
status: 'running',
worldId: worldId,
});
worldStatus = (await ctx.db.get(worldStatusId))!;
await ctx.db.insert('maps', {
worldId,
width: map.mapwidth,
height: map.mapheight,
tileSetUrl: map.tilesetpath,
tileSetDimX: map.tilesetpxw,
tileSetDimY: map.tilesetpxh,
tileDim: map.tiledim,
bgTiles: map.bgtiles,
objectTiles: map.objmap,
decorTiles: map.decors,
bgTilesN: map.bgtilesN,
objectTilesN: map.objmapN,
decorTilesN: map.decorsN,
animatedSprites: map.animatedsprites,
});
await ctx.scheduler.runAfter(0, internal.aiTown.main.runStep, {
worldId,
generationNumber: engine.generationNumber,
maxDuration: ENGINE_ACTION_DURATION,
});
return { worldStatus, engine };
}
async function shouldCreateAgents(
db: DatabaseReader,
worldId: Id<'worlds'>,
engineId: Id<'engines'>,
) {
const world = await db.get(worldId);
if (!world) {
throw new Error(`Invalid world ID: ${worldId}`);
}
if (world.agents.length > 0) {
return false;
}
const unactionedJoinInputs = await db
.query('inputs')
.withIndex('byInputNumber', (q) => q.eq('engineId', engineId))
.order('asc')
.filter((q) => q.eq(q.field('name'), 'createAgent'))
.filter((q) => q.eq(q.field('returnValue'), undefined))
.collect();
if (unactionedJoinInputs.length > 0) {
return false;
}
return true;
}
|