rubber-ducky 1.1.3__tar.gz → 1.1.5__tar.gz

{rubber_ducky-1.1.3 → rubber_ducky-1.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: rubber-ducky
-Version: 1.1.3
+Version: 1.1.5
 Summary: AI Companion for Pair Programming
 Home-page: https://github.com/ParthSareen/ducky
 Author: Parth Sareen
@@ -9,6 +9,14 @@ License: MIT
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ollama
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary
 
 # rubber ducky
 <p align="center">

rubber_ducky-1.1.5/ducky/ducky.py

@@ -0,0 +1,93 @@
+import argparse
+import asyncio
+from ollama import AsyncClient
+
+
+class RubberDuck:
+    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
+
+        if prompt is None:
+            prompt = input("\nEnter your prompt (or press Enter for default review): ")
+
+        if self.quick:
+            prompt += ". Return a command and be extremely concise"
+
+        responses = [self.system_prompt]
+        while True:
+            context_prompt = "\n".join(responses) + "\n" + prompt
+            stream = await self.client.generate(model=self.model, prompt=context_prompt, stream=True)
+            response_text = ""
+            async for chunk in stream:
+                if 'response' in chunk:
+                    print(chunk['response'], end='', flush=True)
+                    response_text += chunk['response']
+            print()
+            responses.append(response_text)
+            if not chain:
+                break
+            prompt = input("\n>> ")
+
+
+def read_files_from_dir(directory: str) -> str:
+    import os
+
+    files = os.listdir(directory)
+    code = ""
+    for file in files:
+        code += open(directory + "/" + file).read()
+    return code
+
+
+async def ducky() -> None:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("question", nargs="*", help="Direct question to ask", default=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",
+        help="Chain the output of the previous command to the next command",
+        action="store_true",
+        default=False,
+    )
+    parser.add_argument(
+        "--model", "-m", help="The model to be used", default="qwen2.5-coder"
+    )
+    args, _ = parser.parse_known_args()
+
+    rubber_ducky = RubberDuck(model=args.model, quick=args.quick)
+
+    # Handle direct question from CLI
+    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:
+        await rubber_ducky.call_llm(prompt=args.prompt)
+        return
+
+    # 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():
+    asyncio.run(ducky())
+
+if __name__ == "__main__":
+    main()

{rubber_ducky-1.1.3 → rubber_ducky-1.1.5}/rubber_ducky.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: rubber-ducky
-Version: 1.1.3
+Version: 1.1.5
 Summary: AI Companion for Pair Programming
 Home-page: https://github.com/ParthSareen/ducky
 Author: Parth Sareen
@@ -9,6 +9,14 @@ License: MIT
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ollama
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary
 
 # rubber ducky
 <p align="center">

{rubber_ducky-1.1.3 → rubber_ducky-1.1.5}/setup.py

@@ -5,7 +5,7 @@ with open('README.md', 'r', encoding='utf-8') as f:
 
 setup(
     name='rubber-ducky',
-    version='1.1.3',
+    version='1.1.5',
     description='AI Companion for Pair Programming',
     long_description=long_description,
     long_description_content_type='text/markdown',

rubber_ducky-1.1.3/ducky/ducky.py

@@ -1,101 +0,0 @@
-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."""
-        self.client = AsyncClient()
-        self.model = model
-
-    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 = []
-        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 = ""
-            async for chunk in stream:
-                if 'response' in chunk:
-                    print(chunk['response'], end='', flush=True)
-                    response_text += chunk['response']
-            print()  # New line after response completes
-            responses.append(response_text)
-            if not chain:
-                break
-            prompt = input("\nAny questions? \n")
-
-
-def read_files_from_dir(directory: str) -> str:
-    import os
-
-    files = os.listdir(directory)
-    code = ""
-    for file in files:
-        code += open(directory + "/" + file).read()
-    return code
-
-
-async def ducky() -> None:
-    parser = argparse.ArgumentParser()
-    parser.add_argument("question", nargs="*", help="Direct question to ask", default=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(
-        "--chain",
-        "-c",
-        help="Chain the output of the previous command to the next command",
-        action="store_true",
-        default=False,
-    )
-    parser.add_argument(
-        "--model", "-m", help="The model to be used", default="codellama"
-    )
-    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)
-
-    # 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)
-        return
-
-    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)
-        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)
-
-
-def main():
-    asyncio.run(ducky())
-
-if __name__ == "__main__":
-    main()