Spaces:

Oldmangrizzz
/

Agent-Node

Paused

Agent-Node / agent_zero_framework /python /api /ctx_window_get.py

Upload folder using huggingface_hub

dff1e71 verified 10 months ago

593 Bytes

from python.helpers.api import ApiHandler, Input, Output, Request



class GetCtxWindow(ApiHandler):
    async def process(self, input: Input, request: Request) -> Output:
        ctxid = input.get("context", [])
        context = self.use_context(ctxid)
        agent = context.streaming_agent or context.agent0
        window = agent.get_data(agent.DATA_NAME_CTX_WINDOW)
        if not window or not isinstance(window, dict):
            return {"content": "", "tokens": 0}

        text = window["text"]
        tokens = window["tokens"]

        return {"content": text, "tokens": tokens}

	from python.helpers.api import ApiHandler, Input, Output, Request



	class GetCtxWindow(ApiHandler):
	async def process(self, input: Input, request: Request) -> Output:
	ctxid = input.get("context", [])
	context = self.use_context(ctxid)
	agent = context.streaming_agent or context.agent0
	window = agent.get_data(agent.DATA_NAME_CTX_WINDOW)
	if not window or not isinstance(window, dict):
	return {"content": "", "tokens": 0}

	text = window["text"]
	tokens = window["tokens"]

	return {"content": text, "tokens": tokens}