# agents/dictionary_agent.py
from agno.agent import Agent
from textwrap import dedent

# Import the specific tool function
from tools.dictionary_tool import check_word_in_dictionary
from tools.segmentation_tool import split_sentences
from tools.morphology_tool import analyze_morphology
# Import model config
from config import SelectedModelClass, SELECTED_MODEL_ID, SELECTED_API_KEY

# Agent definition
DictionaryAgent = None
if SelectedModelClass:
    DictionaryAgent = Agent(
        name="DictionaryExpert",
        # Use the model selected in config.py
        model=SelectedModelClass(id=SELECTED_MODEL_ID, api_key=SELECTED_API_KEY),
        role="Verify if words belong to the official 7000-word list.",
        instructions=dedent("""\
            1. Receive a segment of text.
            2. Use the 'check_word_in_dictionary' tool to identify which words in the text are NOT on the official list.
            3. The tool automatically preprocesses the text (lowercase, splits into words, removes punctuation).
            4. The tool returns a list of non-official words found in the text.
            5. Return this list directly, with no additional comments or formatting.
            6. The expected output is just the list of non-official words. 
            7. If no non-official words are found (the tool returns an empty list), you should still return that empty list as your final answer.
        """),
        tools=[split_sentences, analyze_morphology, check_word_in_dictionary],
        show_tool_calls=False, # Keep False for cleaner production output
        markdown=False, # Output is a list, not markdown text
        # Add memory/storage if needed for context across chunks
    )
else:
    print("ERROR: DictionaryAgent could not be created due to invalid model configuration.")