o չh @sdZddlZddlZddlmZmZddlmZejej de e Z de deedefd d Ze d krWd d ddddddddddgZedeZeejedddSdS)z, Ultra-simple department matching using LLM N)DictList) ChatOllama)level user_prompt departmentsreturnc Csztddd}ddd|D}d|d|d }||j}td |z0d |vrYd |vrY||d |d d }t |}| dd| d| dddWWSWnAt j t fy}z2td||} |D]} | d| vr| d| dddWYd}~WSqsWYd}~nd}~wwddddWSty}ztd|dddt|dWYd}~Sd}~ww)zW Match user goal to department using LLM. Simple. Direct. No overthinking. z gemma3:12bg?)model temperature cSs&g|]}d|dd|ddqS)z- namez (ID: id)).0drrI/var/www/eduai.edurigo.com/doc_train/edurigo_ai/TNA/testing/department.py s&z&analyze_department..zGAnalyze this goal and match it to the most relevant department. Goal: z Available Departments: z Respond in JSON format: {"department_name": "exact name from list", "department_id": number, "confidence": "high/medium/low"} If no good match, use: {"department_name": "", "department_id": null, "confidence": "none"} Response:zLLM response: {}department_name department_id confidencelow)identified_departmentrrz&JSON parse failed, trying text match: r r mediumNnonezAnalysis failed: )rrrerror)rjoininvokecontentstriploggerinfoindexrindexjsonloadsgetJSONDecodeError ValueErrorwarninglower Exceptionrstr) rrllm dept_listpromptresponsejson_strresulteresponse_lowerdeptrrranalyze_department sZ      r:__main__rzHuman Resources)r r zInformation TechnologySales Marketingz6I want to improve team communication and collaboration)indent)__doc__r(loggingtypingrrlangchain_ollamar basicConfigINFO getLogger__name__r$r0r:test_departmentsr6printdumpsrrrrs(  J