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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
ducky/ducky.py CHANGED
@@ -25,9 +25,14 @@ class RubberDuck:
25
25
  while True:
26
26
  # Include previous responses in the prompt for context
27
27
  context_prompt = "\n".join(responses) + "\n" + prompt
28
- response = await self.client.generate(model=self.model, prompt=context_prompt)
29
- print(response['response'])
30
- responses.append(response['response'])
28
+ stream = await self.client.generate(model=self.model, prompt=context_prompt, stream=True)
29
+ response_text = ""
30
+ async for chunk in stream:
31
+ if 'response' in chunk:
32
+ print(chunk['response'], end='', flush=True)
33
+ response_text += chunk['response']
34
+ print() # New line after response completes
35
+ responses.append(response_text)
31
36
  if not chain:
32
37
  break
33
38
  prompt = input("\nAny questions? \n")
@@ -67,7 +72,7 @@ async def ducky() -> None:
67
72
 
68
73
  # Handle direct question from CLI
69
74
  if args.question is not None:
70
- question = " ".join(args.question)
75
+ question = " ".join(args.question) + " be as concise as possible"
71
76
  await rubber_ducky.call_llama(prompt=question, chain=args.chain)
72
77
  return
73
78
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: rubber-ducky
3
- Version: 1.1.2
3
+ Version: 1.1.3
4
4
  Summary: AI Companion for Pair Programming
5
5
  Home-page: https://github.com/ParthSareen/ducky
6
6
  Author: Parth Sareen
@@ -11,6 +11,9 @@ License-File: LICENSE
11
11
  Requires-Dist: ollama
12
12
 
13
13
  # rubber ducky
14
+ <p align="center">
15
+ <img src="ducky_img.webp" alt="Ducky Image" width="200" height="200">
16
+ </p>
14
17
 
15
18
  ## tl;dr
16
19
  - `pip install rubber-ducky`
@@ -0,0 +1,8 @@
1
+ ducky/__init__.py,sha256=_7imP8Jc2SIapn4fzGkspmKvqxPTFDcDJWHZ_o_MnlE,24
2
+ ducky/ducky.py,sha256=OV66MC2-tfimWo5nEBemdKrznFuzp3dOOoJq_P2f2Dk,3859
3
+ rubber_ducky-1.1.3.dist-info/LICENSE,sha256=gQ1rCmw18NqTk5GxG96F6vgyN70e1c4kcKUtWDwdNaE,1069
4
+ rubber_ducky-1.1.3.dist-info/METADATA,sha256=0MKpJ0cTvHulAjWBiwwesXv3Zlvy5p9764Xi2GI-Oq8,1736
5
+ rubber_ducky-1.1.3.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
6
+ rubber_ducky-1.1.3.dist-info/entry_points.txt,sha256=LPndtj8UqEWtwYApv5LJJniH4FUrsriOqV2LA1X_UPQ,43
7
+ rubber_ducky-1.1.3.dist-info/top_level.txt,sha256=4Q75MONDNPpQ3o17bTu7RFuKwFhTIRzlXP3_LDWQQ30,6
8
+ rubber_ducky-1.1.3.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (75.4.0)
2
+ Generator: setuptools (75.5.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -1,8 +0,0 @@
1
- ducky/__init__.py,sha256=_7imP8Jc2SIapn4fzGkspmKvqxPTFDcDJWHZ_o_MnlE,24
2
- ducky/ducky.py,sha256=WhtT9qOIU7xn67Hm9J_F1VPek7brLqqoGmlUlBAl63A,3577
3
- rubber_ducky-1.1.2.dist-info/LICENSE,sha256=gQ1rCmw18NqTk5GxG96F6vgyN70e1c4kcKUtWDwdNaE,1069
4
- rubber_ducky-1.1.2.dist-info/METADATA,sha256=aeTZzl1pw3SAXDDSg7CSh5BN9NRzRaAyjWATkOy9Tjg,1638
5
- rubber_ducky-1.1.2.dist-info/WHEEL,sha256=a7TGlA-5DaHMRrarXjVbQagU3Man_dCnGIWMJr5kRWo,91
6
- rubber_ducky-1.1.2.dist-info/entry_points.txt,sha256=LPndtj8UqEWtwYApv5LJJniH4FUrsriOqV2LA1X_UPQ,43
7
- rubber_ducky-1.1.2.dist-info/top_level.txt,sha256=4Q75MONDNPpQ3o17bTu7RFuKwFhTIRzlXP3_LDWQQ30,6
8
- rubber_ducky-1.1.2.dist-info/RECORD,,