Merge pull request #66 from JoinTheAlliance/compose-state-additional

Add additional key options to composeState

docs/docs/classes/BgentRuntime.md

@@ -179,7 +179,7 @@ ___
 
 ### composeState
 
-▸ **composeState**(`message`): `Promise`\<\{ `actionConditions`: `string` ; `actionExamples`: `string` ; `actionNames`: `string` ; `actions`: `string` ; `actors`: `string` ; `actorsData`: [`Actor`](../interfaces/Actor.md)[] ; `agentId`: \`$\{string}-$\{string}-$\{string}-$\{string}-$\{string}\` ; `agentName`: `undefined` \| `string` ; `evaluatorConditions`: `string` ; `evaluatorExamples`: `string` ; `evaluatorNames`: `string` ; `evaluators`: `string` ; `evaluatorsData`: [`Evaluator`](../interfaces/Evaluator.md)[] ; `goals`: `string` ; `goalsData`: [`Goal`](../interfaces/Goal.md)[] ; `lore`: `string` ; `loreData`: [`Memory`](../interfaces/Memory.md)[] ; `providers`: `string` ; `recentFacts`: `string` ; `recentFactsData`: [`Memory`](../interfaces/Memory.md)[] ; `recentMessages`: `string` ; `recentMessagesData`: [`Memory`](../interfaces/Memory.md)[] ; `relevantFacts`: `string` ; `relevantFactsData`: [`Memory`](../interfaces/Memory.md)[] ; `room_id`: \`$\{string}-$\{string}-$\{string}-$\{string}-$\{string}\` ; `senderName`: `undefined` \| `string`  }\>
+▸ **composeState**(`message`, `additionalKeys?`): `Promise`\<\{ `actionConditions`: `string` ; `actionExamples`: `string` ; `actionNames`: `string` ; `actions`: `string` ; `actors`: `string` ; `actorsData`: [`Actor`](../interfaces/Actor.md)[] ; `agentId`: \`$\{string}-$\{string}-$\{string}-$\{string}-$\{string}\` ; `agentName`: `undefined` \| `string` ; `evaluatorConditions`: `string` ; `evaluatorExamples`: `string` ; `evaluatorNames`: `string` ; `evaluators`: `string` ; `evaluatorsData`: [`Evaluator`](../interfaces/Evaluator.md)[] ; `goals`: `string` ; `goalsData`: [`Goal`](../interfaces/Goal.md)[] ; `lore`: `string` ; `loreData`: [`Memory`](../interfaces/Memory.md)[] ; `providers`: `string` ; `recentFacts`: `string` ; `recentFactsData`: [`Memory`](../interfaces/Memory.md)[] ; `recentMessages`: `string` ; `recentMessagesData`: [`Memory`](../interfaces/Memory.md)[] ; `relevantFacts`: `string` ; `relevantFactsData`: [`Memory`](../interfaces/Memory.md)[] ; `room_id`: \`$\{string}-$\{string}-$\{string}-$\{string}-$\{string}\` ; `senderName`: `undefined` \| `string`  }\>
 
 Compose the state of the agent into an object that can be passed or used for response generation.
 
@@ -188,6 +188,7 @@ Compose the state of the agent into an object that can be passed or used for res
 | Name | Type | Description |
 | :------ | :------ | :------ |
 | `message` | [`Message`](../interfaces/Message.md) | The message to compose the state from. |
+| `additionalKeys` | `Object` | - |
 
 #### Returns
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "bgent",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "private": false,
   "description": "bgent. because agent was taken.",
   "type": "module",

src/lib/__tests__/goals.test.ts

@@ -55,8 +55,6 @@ describe("Goals", () => {
       onlyInProgress: false,
     });
 
-    console.log("goals", goals);
-
     const createdGoal = goals.find((goal: Goal) => goal.name === newGoal.name);
 
     expect(createdGoal).toBeDefined();

src/lib/__tests__/messages.test.ts

@@ -48,7 +48,7 @@ describe("Messages Library", () => {
       runtime,
       room_id,
     });
-    console.log("result", result);
+
     expect(result.length).toBeGreaterThan(0);
     result.forEach((actor: Actor) => {
       expect(actor).toHaveProperty("name");

src/lib/adapters/sqljs.ts

@@ -367,7 +367,6 @@ export class SqlJsDatabaseAdapter extends DatabaseAdapter {
     }
 
     stmt.free();
-    console.log("*** count result:", count);
     return count;
   }
 
@@ -545,7 +544,6 @@ export class SqlJsDatabaseAdapter extends DatabaseAdapter {
     const relationship = stmt.getAsObject() as unknown as
       | Relationship
       | undefined;
-    console.log("relationship", relationship);
     stmt.free();
     return relationship || null;
   }

src/lib/adapters/supabase.ts

@@ -302,7 +302,6 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
   }
 
   async createGoal(goal: Goal): Promise<void> {
-    console.log("inserting goal:", goal);
     const { error } = await this.supabase.from("goals").insert(goal);
     if (error) {
       throw new Error(`Error creating goal: ${error.message}`);

src/lib/runtime.ts

@@ -444,7 +444,10 @@ export class BgentRuntime {
    * @param message The message to compose the state from.
    * @returns The state of the agent.
    */
-  async composeState(message: Message) {
+  async composeState(
+    message: Message,
+    additionalKeys: { [key: string]: unknown } = {},
+  ) {
     const { userId, room_id } = message;
 
     const recentMessageCount = this.getRecentMessageCount();
@@ -549,6 +552,7 @@ export class BgentRuntime {
       recentFactsData,
       relevantFacts: addHeader("# Relevant Facts", relevantFacts),
       relevantFactsData,
+      ...additionalKeys,
     };
 
     const actionPromises = this.actions.map(async (action: Action) => {

src/test/createRuntime.ts

@@ -105,8 +105,6 @@ export async function createRuntime({
             password: TEST_PASSWORD!,
           });
 
-          console.log("response to signup", response);
-
           // Change the name of the user
           const { error } = await supabase
             .from("accounts")

src/test/getOrCreateRelationship.ts

@@ -37,19 +37,14 @@ export async function getOrCreateRelationship({
   }
 
   if (!relationship) {
-    console.log("Creating relationship", userA, userB);
     // Create the relationship
     await runtime.databaseAdapter.createRelationship({
       userA,
       userB,
     });
 
-    console.log("Fetching relationship", userA, userB);
-
     relationship = await getRelationship({ runtime, userA, userB });
 
-    console.log("Fetched relationship", relationship);
-
     if (!relationship) {
       throw new Error("Failed to fetch the created relationship");
     }