rubber-ducky 1.1.1__py3-none-any.whl → 1.1.2__py3-none-any.whl

ducky/ducky.py

@@ -25,7 +25,7 @@ class RubberDuck:
         while True:
             # Include previous responses in the prompt for context
             context_prompt = "\n".join(responses) + "\n" + prompt
-            response = await self.client.generate(model="codellama", prompt=context_prompt)
+            response = await self.client.generate(model=self.model, prompt=context_prompt)
             print(response['response'])
             responses.append(response['response'])
             if not chain:
@@ -45,7 +45,7 @@ def read_files_from_dir(directory: str) -> str:
 
 async def ducky() -> None:
     parser = argparse.ArgumentParser()
-    parser.add_argument("question", nargs="?", help="Direct question to ask", default=None)
+    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)
@@ -67,7 +67,8 @@ async def ducky() -> None:
 
     # Handle direct question from CLI
     if args.question is not None:
-        await rubber_ducky.call_llama(prompt=args.question, chain=args.chain)
+        question = " ".join(args.question)
+        await rubber_ducky.call_llama(prompt=question, chain=args.chain)
         return
 
     if args.file is None and args.directory is None:

{rubber_ducky-1.1.1.dist-info → rubber_ducky-1.1.2.dist-info}/METADATA

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

rubber_ducky-1.1.2.dist-info/RECORD

@@ -0,0 +1,8 @@
+ducky/__init__.py,sha256=_7imP8Jc2SIapn4fzGkspmKvqxPTFDcDJWHZ_o_MnlE,24
+ducky/ducky.py,sha256=WhtT9qOIU7xn67Hm9J_F1VPek7brLqqoGmlUlBAl63A,3577
+rubber_ducky-1.1.2.dist-info/LICENSE,sha256=gQ1rCmw18NqTk5GxG96F6vgyN70e1c4kcKUtWDwdNaE,1069
+rubber_ducky-1.1.2.dist-info/METADATA,sha256=aeTZzl1pw3SAXDDSg7CSh5BN9NRzRaAyjWATkOy9Tjg,1638
+rubber_ducky-1.1.2.dist-info/WHEEL,sha256=a7TGlA-5DaHMRrarXjVbQagU3Man_dCnGIWMJr5kRWo,91
+rubber_ducky-1.1.2.dist-info/entry_points.txt,sha256=LPndtj8UqEWtwYApv5LJJniH4FUrsriOqV2LA1X_UPQ,43
+rubber_ducky-1.1.2.dist-info/top_level.txt,sha256=4Q75MONDNPpQ3o17bTu7RFuKwFhTIRzlXP3_LDWQQ30,6
+rubber_ducky-1.1.2.dist-info/RECORD,,

{rubber_ducky-1.1.1.dist-info → rubber_ducky-1.1.2.dist-info}/WHEEL

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

rubber_ducky-1.1.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-ducky/__init__.py,sha256=_7imP8Jc2SIapn4fzGkspmKvqxPTFDcDJWHZ_o_MnlE,24
-ducky/ducky.py,sha256=NwOzoLMd7fzimq-4OkfEL6buACY2Y13cze2uLEerVBU,3540
-rubber_ducky-1.1.1.dist-info/LICENSE,sha256=gQ1rCmw18NqTk5GxG96F6vgyN70e1c4kcKUtWDwdNaE,1069
-rubber_ducky-1.1.1.dist-info/METADATA,sha256=uPUKC85hZxy2j2jVojKPzWH_rVYt1GAAB7Z1deWUFbk,1638
-rubber_ducky-1.1.1.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-rubber_ducky-1.1.1.dist-info/entry_points.txt,sha256=LPndtj8UqEWtwYApv5LJJniH4FUrsriOqV2LA1X_UPQ,43
-rubber_ducky-1.1.1.dist-info/top_level.txt,sha256=4Q75MONDNPpQ3o17bTu7RFuKwFhTIRzlXP3_LDWQQ30,6
-rubber_ducky-1.1.1.dist-info/RECORD,,