Skip to content

Commit bf74f57

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

2 files changed

Lines changed: 6 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.0',
5+
version='1.1.1',
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: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77
from .package_managers import detect_package_manager
88
from .utils import DB_PATH, save_command_suggestion, fetch_system_info , update_database
99

10+
model = "llama-3.3-70b-versatile"
11+
1012
# Configuration file path
1113
CONFIG_FILE = os.path.expanduser("~/.snapshell_config.json")
1214

@@ -96,7 +98,7 @@ def suggest_command(self, user_input, conversation_history):
9698

9799
chat_completion = self.groq.chat.completions.create(
98100
messages=messages,
99-
model="llama-3.2-90b-text-preview",
101+
model=BaseModel,
100102
temperature=0,
101103
stream=False,
102104
response_format={"type": "json_object"},
@@ -129,7 +131,7 @@ def fallback_to_llm(self, user_input, package_manager, fallback_message, convers
129131

130132
chat_completion = self.groq.chat.completions.create(
131133
messages=messages,
132-
model="llama-3.2-90b-text-preview",
134+
model=BaseModel,
133135
temperature=0,
134136
stream=False,
135137
response_format={"type": "json_object"},
@@ -157,7 +159,7 @@ def formulate_sql_query(self, user_input):
157159

158160
chat_completion = self.groq.chat.completions.create(
159161
messages=messages,
160-
model="llama-3.2-90b-text-preview",
162+
model=BaseModel,
161163
temperature=0,
162164
stream=False,
163165
response_format={"type": "json_object"},

0 commit comments

Comments
 (0)