feat(tool): add produce classification tool

ShrewdAgent.py

@@ -13,6 +13,7 @@ from langgraph.pregel import PregelProtocol
 from loguru import logger
 from pydantic import SecretStr
 
+from tools.produce_classifier import produce_classifier
 from tools.web_page_information_extractor import web_page_information_extractor
 from tools.youtube_transcript import youtube_transcript
 
@@ -40,6 +41,7 @@ class ShrewdAgent:
             TavilySearch(),
             web_page_information_extractor,
             youtube_transcript,
+            produce_classifier,
         ]
         self.llm = ChatOpenAI(
             model="gpt-4o-mini",

tools/produce_classifier.py

@@ -0,0 +1,44 @@
+import os
+
+from langchain_core.tools import tool
+from langchain_openai import ChatOpenAI
+from loguru import logger
+from pydantic import SecretStr
+
+
+@tool("produce_classifier_tool", parse_docstring=True)
+def produce_classifier(food_name: str) -> str:
+    """
+    Classifies a food item as either 'fruit' or 'vegetable' from a botanical perspective.
+
+    Args:
+        food_name (str): The name of the food item to classify.
+
+    Returns:
+        str: The classification of the food item, either 'fruit' or 'vegetable'.
+    """
+    logger.info(f"use produce_classifier_tool with param: {food_name}")
+    chat = ChatOpenAI(
+        model="gpt-4o-mini",
+        temperature=0,
+        api_key = SecretStr(os.environ['OPENAI_API_KEY'])
+    )
+
+    prompt = (f"From a botanical perspective, classify {food_name} as either 'fruit' or 'vegetable'. "
+              #f"If it's not a produce name, classify as 'neither'. "
+              f"Respond only with a JSON in this exact format: "
+              f"{{\"name\": \"{food_name}\", \"kind\": \"[classification]\"}}, "
+              f"where [classification] should be replaced with just 'fruit' or 'vegetable'." #, or 'neither'. "
+              f"No other text or explanation.")
+    return chat.invoke(prompt).content
+
+def _print_produce_kind(food_name: str):
+    print(f'{food_name}: {produce_classifier.invoke(food_name)}')
+
+if __name__ == "__main__":
+    #_print_produce_kind("orange")
+    #_print_produce_kind("sweet potatoes")
+    #_print_produce_kind("bell pepper")
+    #_print_produce_kind("egg")
+    #_print_produce_kind("table")
+    _print_produce_kind("zucchini")