drako 0.1__tar.gz

drako-0.1/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.2
+Name: drako
+Version: 0.1
+Summary: CLI for DrakoAI
+Author: ProjectCuboid
+Dynamic: author
+Dynamic: summary

drako-0.1/drako/cli.py

@@ -0,0 +1,93 @@
+from langchain_ollama import OllamaLLM
+from langchain_core.prompts import ChatPromptTemplate
+
+# Global variable for the user-editable portion of the template.
+editable = """
+You are a friendly AI. You are funny, but dont make it obvius.
+
+Alias: Drako, Gemma, Bot
+"""
+
+def build_template():
+    # This function builds the full prompt template used internally.
+    return editable + """
+Here is the conversation history: {context}
+
+Prompt: {question}
+
+Answer: 
+"""
+
+# Initialize model and chain.
+raw_model = "gemma2:2b"
+model = OllamaLLM(model=raw_model)
+template = build_template()
+prompt = ChatPromptTemplate.from_template(template)
+chain = prompt | model
+
+def handle_conversation():
+    global raw_model, model, editable, template, prompt, chain  # Use globals so we can update them.
+    context = ""
+    print("An Ollama AI model. /? or /help for list of commands.")
+    print("---New conversation---")
+    
+    while True:
+        user_input = input("You: ")
+        
+        if user_input.lower() == "/bye":
+            break
+        
+        elif user_input.lower() in ("/?", "/help"):
+            print(" |  /bye   - for exiting chat.")
+            print(" |  /model - for selecting model. (Not recommended.)")
+            print(" |  /temp  - for viewing the editable template.")
+            print(" |  /edit  - for editing the template.")
+            print(" |  /info  - for information about this chatbot")
+            continue
+        
+        elif user_input.lower() == "/model":
+            print(f"Current model: {raw_model}")
+            raw_model = input(
+                " -WARNING- Changing the model may require installation.\n"
+                "Enter new model to install/select: "
+            )
+            model = OllamaLLM(model=raw_model)  # Update model
+            # Rebuild the chain with the new model.
+            chain = prompt | model
+            continue
+        
+        elif user_input.lower() == "/temp":
+            print("\n--- Editable Template ---")
+            print(editable.strip())
+            print("-------------------------\n")
+            continue
+        
+        elif user_input.lower() == "/edit":
+            new_editable = input("Enter new editable text for the template: ")
+            if new_editable.strip():
+                editable = new_editable
+                template = build_template()
+                prompt = ChatPromptTemplate.from_template(template)
+                chain = prompt | model  # Rebuild the chain with updated template.
+                print("Template updated!")
+            else:
+                print("Template not changed.")
+            continue
+        
+        elif user_input.lower() == "/info":
+            print("\n--- Chatbot Information ---")
+            print("Current Model:", raw_model)
+            # Show only the editable part, not the internal formatting.
+            print("Template (editable part only):")
+            print(editable.strip())
+            print("---------------------------\n")
+            continue
+        
+        try:
+            result = chain.invoke({"context": context, "question": user_input})
+            print("Bot:", result)
+            context += f"\nUser: {user_input}\nAI: {result}"
+        except Exception as e:
+            print("Error:", str(e))
+
+handle_conversation()

drako-0.1/drako/main.py

@@ -0,0 +1,19 @@
+import sys
+from drako.cli import handle_conversation  # Import the conversation handler from chatbot.py
+
+def cli():
+    # Check if a subcommand is provided.
+    if len(sys.argv) < 2:
+        print("Usage: gemma run")
+        sys.exit(1)
+
+    subcommand = sys.argv[1].lower()
+    if subcommand == "run":
+        # Run the chatbot script.
+        handle_conversation()
+    else:
+        print(f"Unknown command: {subcommand}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    cli()

drako-0.1/drako.egg-info/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.2
+Name: drako
+Version: 0.1
+Summary: CLI for DrakoAI
+Author: ProjectCuboid
+Dynamic: author
+Dynamic: summary

drako-0.1/drako.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+setup.py
+drako/__init__.py
+drako/cli.py
+drako/main.py
+drako.egg-info/PKG-INFO
+drako.egg-info/SOURCES.txt
+drako.egg-info/dependency_links.txt
+drako.egg-info/entry_points.txt
+drako.egg-info/top_level.txt

drako-0.1/drako.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

drako-0.1/drako.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+drako = drako.cli:main

drako-0.1/drako.egg-info/top_level.txt

@@ -0,0 +1 @@
+drako

drako-0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

drako-0.1/setup.py

@@ -0,0 +1,21 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="drako",  # The name of your package
+    version="0.1",
+    packages=find_packages(),
+    entry_points={
+        "console_scripts": [
+            # "gemma" is the command you want available after install
+            # "drako.cli:main" is the function that will run when someone types `gemma`
+            "drako = drako.cli:main"
+        ],
+    },
+    install_requires=[
+        # Put your dependencies here, e.g.:
+        # "langchain_ollama",
+        # "langchain_core",
+    ],
+    author="ProjectCuboid",
+    description="CLI for DrakoAI",
+)