Skip to content

Commit 08fbbc9

Browse files
committed
Update create method name
1 parent a3ae36c commit 08fbbc9

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

src/completions/services/completion_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ class CompletionService:
88
def __init__(self, repository: SourceRepository):
99
self.repository = repository
1010

11-
def create_completion(self, query, k=3):
11+
def create(self, query, k=3):
1212
prompt_context = ContextGenerationService(query, self.repository).process(k)
1313

1414
print("\n🧠 Answering with context:\n")

src/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,4 @@
99
q = input("\nAsk a question (or 'exit'): ")
1010
if q.lower() in {"exit", "quit"}:
1111
break
12-
service.create_completion(q)
12+
service.create(q)

tests/completions/services/test_completion_service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@ def service(fake_repository):
2727
"completions.services.completion_service.create_prompt",
2828
return_value="Generated Prompt",
2929
)
30-
def test_create_completion(mock_create_prompt, mock_create_completion, service, capsys):
30+
def test_create(mock_create_prompt, mock_create_completion, service, capsys):
3131
"""Should call the LLM with the correct prompt and print the answer."""
3232

33-
service.create_completion("What is AI?", k=3)
33+
service.create("What is AI?", k=3)
3434

3535
service.repository.get_top_k_chunks_by_similarity.assert_called_once_with(
3636
"What is AI?", 3

0 commit comments

Comments
 (0)