mirror of
https://github.com/kennethnym/freya
synced 2026-06-23 09:55:09 +01:00
feat: add exa web search source (#124)
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
import type { ActionDefinition } from "@freya/core"
|
||||
import type { Context, Hono } from "hono"
|
||||
|
||||
import { type } from "arktype"
|
||||
@@ -55,6 +56,13 @@ export function registerSourcesHttpHandlers(
|
||||
app.get("/api/sources/:sourceId", inject, authSessionMiddleware, handleGetSource)
|
||||
app.patch("/api/sources/:sourceId", inject, authSessionMiddleware, handleUpdateSource)
|
||||
app.put("/api/sources/:sourceId", inject, authSessionMiddleware, handleReplaceSource)
|
||||
app.get("/api/sources/:sourceId/actions", inject, authSessionMiddleware, handleListActions)
|
||||
app.post(
|
||||
"/api/sources/:sourceId/actions/:actionId",
|
||||
inject,
|
||||
authSessionMiddleware,
|
||||
handleExecuteAction,
|
||||
)
|
||||
app.put(
|
||||
"/api/sources/:sourceId/credentials",
|
||||
inject,
|
||||
@@ -189,6 +197,71 @@ async function handleReplaceSource(c: Context<Env>) {
|
||||
return c.body(null, 204)
|
||||
}
|
||||
|
||||
async function handleListActions(c: Context<Env>) {
|
||||
const sourceId = c.req.param("sourceId")
|
||||
if (!sourceId) {
|
||||
return c.body(null, 404)
|
||||
}
|
||||
|
||||
const user = c.get("user")!
|
||||
const sessionManager = c.get("sessionManager")
|
||||
|
||||
let session
|
||||
try {
|
||||
session = await sessionManager.getOrCreate(user.id)
|
||||
} catch (err) {
|
||||
console.error("[handleListActions] Failed to create session:", err)
|
||||
return c.json({ error: "Service unavailable" }, 503)
|
||||
}
|
||||
|
||||
try {
|
||||
const actions = await session.engine.listActions(sourceId)
|
||||
return c.json({ actions: serializeActions(actions) })
|
||||
} catch (err) {
|
||||
if (isActionNotFoundError(err)) {
|
||||
return c.json({ error: err.message }, 404)
|
||||
}
|
||||
console.error(`[handleListActions] Failed to list actions for "${sourceId}":`, err)
|
||||
return c.json({ error: "Failed to list actions" }, 500)
|
||||
}
|
||||
}
|
||||
|
||||
async function handleExecuteAction(c: Context<Env>) {
|
||||
const sourceId = c.req.param("sourceId")
|
||||
const actionId = c.req.param("actionId")
|
||||
if (!sourceId || !actionId) {
|
||||
return c.body(null, 404)
|
||||
}
|
||||
|
||||
let params: unknown
|
||||
try {
|
||||
params = await c.req.json()
|
||||
} catch {
|
||||
return c.json({ error: "Invalid JSON" }, 400)
|
||||
}
|
||||
|
||||
const user = c.get("user")!
|
||||
const sessionManager = c.get("sessionManager")
|
||||
|
||||
let session
|
||||
try {
|
||||
session = await sessionManager.getOrCreate(user.id)
|
||||
} catch (err) {
|
||||
console.error("[handleExecuteAction] Failed to create session:", err)
|
||||
return c.json({ error: "Service unavailable" }, 503)
|
||||
}
|
||||
|
||||
try {
|
||||
const result = await session.engine.executeAction(sourceId, actionId, params)
|
||||
return c.json({ result })
|
||||
} catch (err) {
|
||||
if (isActionNotFoundError(err)) {
|
||||
return c.json({ error: err.message }, 404)
|
||||
}
|
||||
return c.json({ error: err instanceof Error ? err.message : String(err) }, 400)
|
||||
}
|
||||
}
|
||||
|
||||
async function handleUpdateCredentials(c: Context<Env>) {
|
||||
const sourceId = c.req.param("sourceId")
|
||||
if (!sourceId) {
|
||||
@@ -228,3 +301,21 @@ async function handleUpdateCredentials(c: Context<Env>) {
|
||||
|
||||
return c.body(null, 204)
|
||||
}
|
||||
|
||||
function serializeActions(actions: Record<string, ActionDefinition>) {
|
||||
const serialized: Record<string, { id: string; description?: string }> = {}
|
||||
for (const [key, action] of Object.entries(actions)) {
|
||||
serialized[key] = {
|
||||
id: action.id,
|
||||
...(action.description ? { description: action.description } : {}),
|
||||
}
|
||||
}
|
||||
return serialized
|
||||
}
|
||||
|
||||
function isActionNotFoundError(err: unknown): err is Error {
|
||||
if (!(err instanceof Error)) {
|
||||
return false
|
||||
}
|
||||
return err.message.startsWith("Source not found:") || err.message.startsWith("Action ")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user