Skip to content

[hackathon] UDF Copilot: schema-aware AI in the Python UDF editor #5081

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
mengw15 wants to merge 3 commits into
base: main
Choose a base branch
from
Open

[hackathon] UDF Copilot: schema-aware AI in the Python UDF editor #5081

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
0d9a128
init
mengw15 May 15, 2026
0526172
init 2
mengw15 May 15, 2026
77f71cf
chore: remove personal config accidentally committed earlier
mengw15 May 15, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions agent-service/src/agent/prompts.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ import { WorkflowSystemMetadata } from "./util/workflow-system-metadata";
const PYTHON_UDF_OPERATOR_TYPES = ["PythonUDFV2"];
const R_UDF_OPERATOR_TYPES = ["RUDF"];

const PYTHON_UDF_INSTRUCTIONS = `## Python UDF Guide
export const PYTHON_UDF_INSTRUCTIONS = `## Python UDF Guide

Python UDF operators run user-defined Python code. There are 2 APIs to process data:

Expand Down Expand Up @@ -83,7 +83,11 @@ class ProcessTableOperator(UDFTableOperator):

- DO NOT change the class name (ProcessTupleOperator or ProcessTableOperator).
- Import packages explicitly (pandas, numpy, etc.).
- Tuple is a Python dict. Access fields with tuple_["field"] ONLY (no .get/.set/.values).
- **Tuple is NOT a Python dict.** Access fields with \`tuple_["field"]\` for reads and writes ONLY. Dict-style methods like \`.items()\`, \`.keys()\`, \`.values()\`, \`.get()\` DO NOT exist on Tuple and will raise AttributeError.
- To iterate fields: \`for name, value in tuple_.as_key_value_pairs():\` (preferred) or \`for name in tuple_.get_field_names(): value = tuple_[name]\`.
- To convert to a real dict: \`tuple_.as_dict()\` returns an OrderedDict.
- To convert to a pandas Series: \`tuple_.as_series()\`.
- \`len(tuple_)\` and \`name in tuple_\` work as expected.
- Table is a pandas DataFrame.
- Use yield to return results.
- Handle None values carefully.
Expand Down
2 changes: 2 additions & 0 deletions agent-service/src/server.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ import { Elysia, t } from "elysia";
import { cors } from "@elysiajs/cors";
import { createOpenAI } from "@ai-sdk/openai";
import { TexeraAgent } from "./agent/texera-agent";
import { udfCopilotRouter } from "./udf-copilot/router";
import { getBackendConfig } from "./api/backend-api";
import { extractUserFromToken, validateToken } from "./api/auth-api";
import { retrieveWorkflow } from "./api/workflow-api";
Expand Down Expand Up @@ -484,6 +485,7 @@ export function buildApp() {
timestamp: new Date().toISOString(),
}))
.use(agentsRouter)
.use(udfCopilotRouter)
)
.ws(`${env.API_PREFIX}/agents/:id/react`, {
open(ws) {
Expand Down
Loading
Loading