fix(js-sdk): Got testing suite working with current sdk (#111)

AI-Engineer-Foundation · Apr 2, 2024 · 1f00196 · 1f00196
1 parent e2edf1d
commit 1f00196
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 9 deletions.
diff --git a/packages/sdk/js/examples/minimal.ts b/packages/sdk/js/examples/minimal.ts
@@ -18,4 +18,4 @@ async function taskHandler(taskInput: TaskInput | null): Promise<StepHandler> {
  return stepHandler
 }
 
-Agent.handleTask(taskHandler).start()
+Agent.handleTask(taskHandler, {}).start()
diff --git a/packages/sdk/js/src/agent.ts b/packages/sdk/js/src/agent.ts
@@ -65,6 +65,7 @@ export const createAgentTask = async (
  const task: Task = {
  task_id: uuid(),
  input: body?.input ?? null,
+ additional_input: body?.additional_input ?? {},
  artifacts: [],
  }
  const stepHandler = await taskHandler(task.task_id, body?.input ?? null)
@@ -89,15 +90,47 @@ const registerCreateAgentTask: RouteRegisterFn = (router: Router) => {
  * Lists all tasks that have been created for the agent.
  * @returns Promise<string[]>
  */
-export const listAgentTaskIDs = async (): Promise<string[]> => {
- return tasks.map(([task, _]) => task.task_id)
+export const listAgentTasks = async (): Promise<Task[]> => {
+ return tasks.map(([task, _]) => task)
 }
-const registerListAgentTaskIDs: RouteRegisterFn = (router: Router) => {
+const registerListAgentTasks: RouteRegisterFn = (router: Router) => {
  router.get('/agent/tasks', (req, res) => {
  void (async () => {
  try {
- const ids = await listAgentTaskIDs()
- res.status(200).json(ids)
+ const tasks = await listAgentTasks()
+
+ let currentPage = 1
+ let pageSize = 10
+ if (
+ req.query.current_page !== undefined &&
+ !isNaN(Number(req.query.current_page))
+ ) {
+ currentPage = Number(req.query.current_page)
+ }
+
+ if (
+ req.query.page_size !== undefined &&
+ !isNaN(Number(req.query.page_size))
+ ) {
+ pageSize = Number(req.query.page_size)
+ }
+
+ const totalItems = tasks.length
+ const totalPages = Math.ceil(totalItems / pageSize)
+
+ const start = (currentPage - 1) * pageSize
+ const end = start + pageSize
+ const pagedTasks = tasks.slice(start, end)
+
+ res.status(200).json({
+ tasks: pagedTasks,
+ pagination: {
+ total_items: totalItems,
+ total_pages: totalPages,
+ current_page: currentPage,
+ page_size: pageSize,
+ }
+ })
  } catch (err: Error | any) {
  console.error(err)
  res.status(500).json({ error: err.message })
@@ -471,7 +504,7 @@ export class Agent {
  port: port ?? this.config.port ?? defaultAgentConfig.port,
  routes: [
  registerCreateAgentTask,
- registerListAgentTaskIDs,
+ registerListAgentTasks,
  registerGetAgentTask,
  registerListAgentTaskSteps,
  registerExecuteAgentTaskStep,

diff --git a/packages/sdk/js/src/index.ts b/packages/sdk/js/src/index.ts
@@ -13,7 +13,7 @@ import {
  type TaskHandler,
  StepResultWithDefaults,
  createAgentTask,
- listAgentTaskIDs,
+ listAgentTasks,
  getAgentTask,
  listAgentTaskSteps,
  executeAgentTaskStep,
@@ -34,7 +34,7 @@ export {
  type TaskHandler,
  StepResultWithDefaults as StepResult,
  createAgentTask,
- listAgentTaskIDs,
+ listAgentTasks,
  getAgentTask,
  listAgentTaskSteps,
  executeAgentTaskStep,

diff --git a/packages/sdk/js/src/models.ts b/packages/sdk/js/src/models.ts
@@ -2,6 +2,7 @@
  * Input parameters for the task. Any value is allowed.
  */
 export type TaskInput = any
+export type AdditionalInput = any
 
 /**
  * Artifact that the task has produced. Any value is allowed.
@@ -87,6 +88,7 @@ export interface StepResult {
 
 export interface Task {
  input?: TaskInput
+ additional_input?: AdditionalInput
  /**
  * The ID of the task.
  */
@@ -99,4 +101,5 @@ export interface Task {
 
 export interface TaskRequestBody {
  input?: TaskInput
+ additional_input?: AdditionalInput
 }