Function get_prompt

AskAgent/python/chatbot_interface.py:199–245

Forward get_prompt to NLWeb

(name: str, arguments: dict | None)

    @server.get_prompt()
    async def get_prompt(name: str, arguments: dict | None) -> GetPromptResult:
        """Forward get_prompt to NLWeb"""
        if not arguments:
            arguments = {}

        # Add the prompt name as prompt_id if not present
        if "prompt_id" not in arguments:
            arguments["prompt_id"] = name

        result = await forward_to_nlweb("get_prompt", arguments, server_url, endpoint)

        if "error" in result:
            return GetPromptResult(
                description=f"Failed to get prompt {arguments.get('prompt_id', name)}",
                messages=[
                    PromptMessage(
                        role="user",
                        content=TextContent(type="text", text=f"Error: {result['error']}")
                    )
                ]
            )

        # Extract prompt from the response
        try:
            prompt_data = result.get("response", {})
            prompt_text = prompt_data.get("prompt_text", f"Prompt for {name}")

            return GetPromptResult(
                description=f"Prompt: {prompt_data.get('name', name)}",
                messages=[
                    PromptMessage(
                        role="user",
                        content=TextContent(type="text", text=prompt_text)
                    )
                ]
            )
        except Exception as e:
            print(f"Error processing prompt: {e!s}", file=sys.stderr)
            return GetPromptResult(
                description=f"Error getting prompt {name}",
                messages=[
                    PromptMessage(
                        role="user",
                        content=TextContent(type="text", text=f"Error: {e!s}")
                    )
                ]
            )

    # Run the server
    options = server.create_initialization_options()