Skip to content

Commit dc2e2a5

Browse files
committed
Change: Model change
1 parent bf74f57 commit dc2e2a5

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
setup(
44
name='snapshell',
5-
version='1.1.1',
5+
version='1.1.2',
66
author='Krishnatejaswi S',
77
author_email='shentharkrishnatejaswi@gmail.com',
88
description='An AI-powered tool to suggest Linux commands based on user input',

snapshell/llm_api.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def suggest_command(self, user_input, conversation_history):
9898

9999
chat_completion = self.groq.chat.completions.create(
100100
messages=messages,
101-
model=BaseModel,
101+
model=model,
102102
temperature=0,
103103
stream=False,
104104
response_format={"type": "json_object"},
@@ -131,7 +131,7 @@ def fallback_to_llm(self, user_input, package_manager, fallback_message, convers
131131

132132
chat_completion = self.groq.chat.completions.create(
133133
messages=messages,
134-
model=BaseModel,
134+
model=model,
135135
temperature=0,
136136
stream=False,
137137
response_format={"type": "json_object"},
@@ -159,7 +159,7 @@ def formulate_sql_query(self, user_input):
159159

160160
chat_completion = self.groq.chat.completions.create(
161161
messages=messages,
162-
model=BaseModel,
162+
model=model,
163163
temperature=0,
164164
stream=False,
165165
response_format={"type": "json_object"},

0 commit comments

Comments
 (0)