Source code for aidputils.agents.toolkit.configs

from pydantic import BaseModel
from typing import Optional, Union, Dict, Any
import uuid 



[docs]
class ModelArgs(BaseModel):
    temperature: Optional[float] = 1.0
    max_tokens: Optional[int] = 600
    frequency_penalty: Optional[int] = 0 
    presence_penalty: Optional[int] = 0
    top_k: Optional[int] = 1 
    top_p: Optional[float] = 0.75 





[docs]
class OCIAIConf(BaseModel):
    model_id: str
    model_provider: Optional[str] =  None
    model_args: Optional[dict] = {}
    endpoint: str
    compartment_id: str
    auth_type: Optional[str] = "remote"
    auth_profile: Optional[str] = "DEFAULT"
    system_prompt: Optional[str] = None
    guardrails_config: Optional[Union[str, Dict, Any]] = None






[docs]
class AIDPToolConf(BaseModel):
    name: str
    description: Optional[str]
    tool_class: str
    conf: dict
    params: Optional[list]





[docs]
class InvokeConfig(BaseModel):
    thread_id: Optional[str] = uuid.uuid4()
    checkpoint_ns: Optional[str] = "default"
    dh_user_principal: Optional[str] = None
    session_variables: Optional[dict] = None





[docs]
class LangGraphRunnableConfig(BaseModel):
    configurable: InvokeConfig