refactor: file renaming etc. Default is graph agent.

core/Agents/neoagent.py → core/Agents/neo_agent.py

@@ -25,7 +25,7 @@
 ReAct is a simple multi-step agent architecture.
 Smaller graphs are often better understood by the LLMs.
 """
-class Neoagent:
+class NeoAgent:
     def __init__(self):
         print("""
 ------------------------------

core/graphAgent.py

@@ -19,6 +19,11 @@
 
 class Graph:
     def __init__(self):
+        print("""
+------------------------------
+Instantiated Graph Agent....
+------------------------------
+            """)
         self.workflow = StateGraph(GraphState)
 
         self.workflow.add_node("jarvis_agent", jarvis_agent)

core/main.py

@@ -1,6 +1,6 @@
 from flask import Flask, request, url_for, jsonify
 from graphAgent import Graph
-from Agents.neoagent import Neoagent
+from core.Agents.neo_agent import NeoAgent
 from models import Model
 from summarize_chat import summarize_chat
 from rag import embed_and_store
@@ -31,9 +31,9 @@
 
 # Agent instantiation
 # Graph() contains all complex tools
-# Neoagent() is a simple ReAct agent that only has websearch and the add tool. For testing purposes.
+# NeoAgent() is a simple ReAct agent that only has websearch and the add tool. For testing purposes.
 jarvis = Graph() # API key is configured in agent.py
-#jarvis = Neoagent()
+#jarvis = NeoAgent()
 
 # Initialize active_chats with the correct format
 active_chats = defaultdict(lambda: {"chat_history": []})