Update myagent.py

myagent.py

@@ -10,7 +10,7 @@ import torch
 # --- Basic Agent Definition ---
 
 # Basic model wrapper for local inference with debug info
-class LocalLlamaModel:
+class BasicAgent:
     def __init__(self, model, tokenizer):
         self.model = model
         self.tokenizer = tokenizer
@@ -74,7 +74,7 @@ def load_model(model_id="LiquidAI/LFM2-1.2B"):
         trust_remote_code=True,
     )
     tokenizer = AutoTokenizer.from_pretrained(model_id)
-    return LocalLlamaModel(model, tokenizer)
+    return BasicAgent(model, tokenizer)
 
 # Run minimal test
 if __name__ == "__main__":