(no commit message)

This commit is contained in:
2025-10-25 03:28:44 -04:00
parent c1462fa1f8
commit 655ded986e
2 changed files with 116 additions and 14 deletions

18
main.py
View File

@@ -1,15 +1,13 @@
from modaic import PrecompiledAgent, PrecompiledConfig
import dspy
class ReceiptClassifierConfig(
PrecompiledConfig
): # configurable for each agent instance
lm: str = "openai/gpt-4o-mini" # set OPENAI_API_KEY
): # Configurable for each agent instance
lm: str = "openai/gpt-4o-mini" # Set OPENAI_API_KEY
max_tokens: int = 500
class ReceiptClassifier(PrecompiledAgent): # extends dspy.module()
class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module()
config: ReceiptClassifierConfig
def __init__(self, config: ReceiptClassifierConfig, **kwargs):
@@ -23,13 +21,5 @@ class ReceiptClassifier(PrecompiledAgent): # extends dspy.module()
receipt_image = dspy.Image(url=image_url)
return self.extract(image=receipt_image)
agent = ReceiptClassifier(ReceiptClassifierConfig())
def main():
agent.push_to_hub("farouk1/receipt-classifier", with_code=True)
if __name__ == "__main__":
main()
agent.push_to_hub("farouk1/receipt-classifier", with_code=True) # Set MODAIC_TOKEN