Custom Agent Dynamic SELECT Param
Hello Dataiku Team,
I am creating a custom agent, and I am trying to add a python script for the SELECT parameter, but it is not working.
Here is what I have in my agent.json:
Here is how my tree looks like:
And the compute_llm_choices.py is like this:
Please, could you spot any reason why this is not working from the snippets I shared?
I would think a custom agent works similar to a plugin, so this procedure should work?
Thank you
Answers
-
JordanG Dataiker, Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Adv Designer, Registered Posts: 14 DataikerHey @csanti — Is there a reason you're attempting to do this via a Plugin rather than the LLM Mesh? I don't know much about your use case but choosing different LLMs is a core component of the LLM Mesh. Within the LLM Mesh, you are able to create visual Agents along with custom code Agents and tools to facilitate Agentic use cases. Depending on your current license, some of these features may not be present until later this fall.
-
areisdorf Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Adv Designer, Registered Posts: 6 ✭✭✭✭
This is also not working for me. Would be good to be able to provide the user a list of their available models instead of having to hardcode the values.
-
csanti Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Adv Designer, Registered Posts: 8 ✭✭✭
Sorry for the delay to reply. Somehow I missed this response.
I am using LLM Mesh. I am attempting to be able to select the LLM from the agent UI.
Using a dynamic select is available for regular plugins. Since the API is similar for agents (JSON + PY file, with similar fields as used in the plugins, such as SELECT), I am trying to understand how this functionality is not available for agents.