rubber-ducky 1.1.3__py3-none-any.whl → 1.1.4__py3-none-any.whl

ducky/ducky.py

@@ -1,29 +1,29 @@
 import argparse
 import asyncio
-from typing import Optional
 from ollama import AsyncClient
 
 
 class RubberDuck:
-    def __init__(self, model: str = "codellama") -> None:
-        self.system_prompt = """You are a pair progamming tool to help developers debug, think through design, and write code. 
-        Help the user think through their approach and provide feedback on the code. Think step by step and ask clarifying questions if needed."""
+    def __init__(self, model: str, quick: bool = False) -> None:
+        self.system_prompt = """You are a pair progamming tool called Ducky or RubberDucky to help developers debug, think through design, and write code. 
+        Help the user think through their approach and provide feedback on the code. Think step by step and ask clarifying questions if needed.
+        If asked """
         self.client = AsyncClient()
         self.model = model
+        self.quick = quick
+
+    async def call_llm(self, prompt: str | None = None) -> None:
+        chain = False if prompt else True
 
-    async def call_llama(self, code: str = "", prompt: Optional[str] = None, chain: bool = False) -> None:
         if prompt is None:
-            user_prompt = input("\nEnter your prompt (or press Enter for default review): ")
-            if not user_prompt:
-                prompt = "review the code, find any issues if any, suggest cleanups if any:" + code
-            else:
-                prompt = user_prompt + code
-        else:
-            prompt = prompt + code
-
-        responses = []
+            prompt = input("\nEnter your prompt (or press Enter for default review): ")
+
+        if self.quick:
+            prompt += ". Return a command and be extremely concise"
+
+        print(prompt)
+        responses = [self.system_prompt]
         while True:
-            # Include previous responses in the prompt for context
             context_prompt = "\n".join(responses) + "\n" + prompt
             stream = await self.client.generate(model=self.model, prompt=context_prompt, stream=True)
             response_text = ""
@@ -31,11 +31,11 @@ class RubberDuck:
                 if 'response' in chunk:
                     print(chunk['response'], end='', flush=True)
                     response_text += chunk['response']
-            print()  # New line after response completes
+            print()
             responses.append(response_text)
             if not chain:
                 break
-            prompt = input("\nAny questions? \n")
+            prompt = input("\n>> ")
 
 
 def read_files_from_dir(directory: str) -> str:
@@ -54,6 +54,7 @@ async def ducky() -> None:
     parser.add_argument("--prompt", "-p", help="Custom prompt to be used", default=None)
     parser.add_argument("--file", "-f", help="The file to be processed", default=None)
     parser.add_argument("--directory", "-d", help="The directory to be processed", default=None)
+    parser.add_argument("--quick", "-q", help="Quick mode", default=False)
     parser.add_argument(
         "--chain",
         "-c",
@@ -62,36 +63,29 @@ async def ducky() -> None:
         default=False,
     )
     parser.add_argument(
-        "--model", "-m", help="The model to be used", default="codellama"
+        "--model", "-m", help="The model to be used", default="qwen2.5-coder"
     )
     args, _ = parser.parse_known_args()
 
-    # My testing has shown that the codellama:7b-python is good for returning python code from the program.
-    # My intention with this tool was to give more general feedback and have back a back and forth with the user.
-    rubber_ducky = RubberDuck(model=args.model)
+    rubber_ducky = RubberDuck(model=args.model, quick=args.quick)
 
     # Handle direct question from CLI
-    if args.question is not None:
-        question = " ".join(args.question) + " be as concise as possible"
-        await rubber_ducky.call_llama(prompt=question, chain=args.chain)
+    print(args.question)
+    if args.question:
+        question = " ".join(args.question) 
+        await rubber_ducky.call_llm(prompt=question)
         return
 
+    # Handle interactive mode (no file/directory specified)
     if args.file is None and args.directory is None:
-        # Handle interactive mode (no file/directory specified)
-        await rubber_ducky.call_llama(prompt=args.prompt, chain=args.chain)
-        if args.chain:
-            while True:
-                await rubber_ducky.call_llama(prompt=args.prompt, chain=args.chain)
+        await rubber_ducky.call_llm(prompt=args.prompt)
         return
 
-    # Handle file input
-    if args.file is not None:
-        code = open(args.file).read()
-    # Handle directory input
-    else:
-        code = read_files_from_dir(args.directory)
-        
-    await rubber_ducky.call_llama(code=code, prompt=args.prompt, chain=args.chain)
+    # Get code from file or directory
+    code = (open(args.file).read() if args.file 
+            else read_files_from_dir(args.directory))
+
+    await rubber_ducky.call_llm(code=code, prompt=args.prompt)
 
 
 def main():

{rubber_ducky-1.1.3.dist-info → rubber_ducky-1.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rubber-ducky
-Version: 1.1.3
+Version: 1.1.4
 Summary: AI Companion for Pair Programming
 Home-page: https://github.com/ParthSareen/ducky
 Author: Parth Sareen

rubber_ducky-1.1.4.dist-info/RECORD

@@ -0,0 +1,8 @@
+ducky/__init__.py,sha256=_7imP8Jc2SIapn4fzGkspmKvqxPTFDcDJWHZ_o_MnlE,24
+ducky/ducky.py,sha256=wJZMwjyw3JQnNcKwOdhO05l86OUfL1iEVnkBkvYGPss,3292
+rubber_ducky-1.1.4.dist-info/LICENSE,sha256=gQ1rCmw18NqTk5GxG96F6vgyN70e1c4kcKUtWDwdNaE,1069
+rubber_ducky-1.1.4.dist-info/METADATA,sha256=CCn5Tdj5miUeDSELtNVe8y85btA0XbZkcey-gs6Orpk,1736
+rubber_ducky-1.1.4.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+rubber_ducky-1.1.4.dist-info/entry_points.txt,sha256=LPndtj8UqEWtwYApv5LJJniH4FUrsriOqV2LA1X_UPQ,43
+rubber_ducky-1.1.4.dist-info/top_level.txt,sha256=4Q75MONDNPpQ3o17bTu7RFuKwFhTIRzlXP3_LDWQQ30,6
+rubber_ducky-1.1.4.dist-info/RECORD,,

{rubber_ducky-1.1.3.dist-info → rubber_ducky-1.1.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.5.0)
+Generator: setuptools (75.7.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

rubber_ducky-1.1.3.dist-info/RECORD

@@ -1,8 +0,0 @@
-ducky/__init__.py,sha256=_7imP8Jc2SIapn4fzGkspmKvqxPTFDcDJWHZ_o_MnlE,24
-ducky/ducky.py,sha256=OV66MC2-tfimWo5nEBemdKrznFuzp3dOOoJq_P2f2Dk,3859
-rubber_ducky-1.1.3.dist-info/LICENSE,sha256=gQ1rCmw18NqTk5GxG96F6vgyN70e1c4kcKUtWDwdNaE,1069
-rubber_ducky-1.1.3.dist-info/METADATA,sha256=0MKpJ0cTvHulAjWBiwwesXv3Zlvy5p9764Xi2GI-Oq8,1736
-rubber_ducky-1.1.3.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-rubber_ducky-1.1.3.dist-info/entry_points.txt,sha256=LPndtj8UqEWtwYApv5LJJniH4FUrsriOqV2LA1X_UPQ,43
-rubber_ducky-1.1.3.dist-info/top_level.txt,sha256=4Q75MONDNPpQ3o17bTu7RFuKwFhTIRzlXP3_LDWQQ30,6
-rubber_ducky-1.1.3.dist-info/RECORD,,