naman1102 commited on
Commit
27f4250
·
1 Parent(s): 791be58

Update analyzer.py

Browse files
Files changed (1) hide show
  1. analyzer.py +5 -3
analyzer.py CHANGED
@@ -6,9 +6,11 @@ def analyze_code(code: str) -> str:
6
  Uses OpenAI's GPT-4.1 mini model to analyze the given code.
7
  Returns the analysis as a string.
8
  """
 
 
9
  system_prompt = "You are a helpful assistant. Analyze the code given to you. Provide insights, strengths, weaknesses, and suggestions for improvement."
10
- response = openai.ChatCompletion.create(
11
- model="gpt-4.1-mini", # GPT-4.1 mini
12
  messages=[
13
  {"role": "system", "content": system_prompt},
14
  {"role": "user", "content": code}
@@ -16,7 +18,7 @@ def analyze_code(code: str) -> str:
16
  max_tokens=512,
17
  temperature=0.7
18
  )
19
- return response.choices[0].message["content"]
20
 
21
  def combine_repo_files_for_llm(repo_dir="repo_files", output_file="combined_repo.txt"):
22
  """
 
6
  Uses OpenAI's GPT-4.1 mini model to analyze the given code.
7
  Returns the analysis as a string.
8
  """
9
+ from openai import OpenAI
10
+ client = OpenAI()
11
  system_prompt = "You are a helpful assistant. Analyze the code given to you. Provide insights, strengths, weaknesses, and suggestions for improvement."
12
+ response = client.chat.completions.create(
13
+ model="gpt-4-1106-preview", # GPT-4.1 mini
14
  messages=[
15
  {"role": "system", "content": system_prompt},
16
  {"role": "user", "content": code}
 
18
  max_tokens=512,
19
  temperature=0.7
20
  )
21
+ return response.choices[0].message.content
22
 
23
  def combine_repo_files_for_llm(repo_dir="repo_files", output_file="combined_repo.txt"):
24
  """