You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from __future__ import annotations
|
|
from typing import Optional
|
|
from agentkit.context import ConversationContext
|
|
from agentkit.types import PluginConfigParam, OnProgressCallback
|
|
|
|
class AgentKitPostprocessor:
|
|
CONFIG: list[PluginConfigParam] = []
|
|
INPUT: list[PluginConfigParam] = []
|
|
OUTPUT: list[PluginConfigParam] = []
|
|
|
|
def __init__(self, props: dict, **kwargs):
|
|
pass
|
|
|
|
async def __aenter__(self):
|
|
return self
|
|
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
|
return None
|
|
|
|
async def on_create_conversation(self, on_progress: Optional[OnProgressCallback] = None) -> None:
|
|
return None
|
|
|
|
async def on_open_conversation(self, on_progress: Optional[OnProgressCallback] = None) -> None:
|
|
return None
|
|
|
|
async def on_after_completion(self, context: ConversationContext, **kwargs) -> Optional[dict]:
|
|
"""
|
|
Called after the conversation, before the completion is returned to the user.
|
|
"""
|
|
return None
|
|
|
|
async def on_after_completion_background(self, context: ConversationContext, **kwargs) -> Optional[dict]:
|
|
"""
|
|
Called after the conversation, after the completion is returned to the user.
|
|
"""
|
|
return None |