AdityaDevx commited on
Commit
5fd1110
·
1 Parent(s): afb53f4

fix: better error handling, reduce truncation to 6000 chars

Browse files
Files changed (1) hide show
  1. api.py +13 -10
api.py CHANGED
@@ -40,8 +40,8 @@ def run_scan(owner, repo, file_path):
40
  return {"error": err}
41
 
42
  # Truncate large files
43
- if len(code) > 8000:
44
- code = code[:8000] + "\n... [truncated]"
45
 
46
  client = Groq(api_key=os.getenv("GROQ_API_KEY", ""))
47
 
@@ -81,14 +81,17 @@ Provide a detailed security analysis in this exact markdown format:
81
  - **Overall Risk**: [CRITICAL/HIGH/MEDIUM/LOW]
82
  """
83
 
84
- response = client.chat.completions.create(
85
- model="llama-3.3-70b-versatile",
86
- messages=[{"role": "user", "content": prompt}],
87
- temperature=0.1,
88
- max_tokens=4096,
89
- )
90
- result = response.choices[0].message.content
91
- return {"result": result}
 
 
 
92
 
93
 
94
  class ScanRequest(BaseModel):
 
40
  return {"error": err}
41
 
42
  # Truncate large files
43
+ if len(code) > 6000:
44
+ code = code[:6000] + "\n... [truncated]"
45
 
46
  client = Groq(api_key=os.getenv("GROQ_API_KEY", ""))
47
 
 
81
  - **Overall Risk**: [CRITICAL/HIGH/MEDIUM/LOW]
82
  """
83
 
84
+ try:
85
+ response = client.chat.completions.create(
86
+ model="llama-3.3-70b-versatile",
87
+ messages=[{"role": "user", "content": prompt}],
88
+ temperature=0.1,
89
+ max_tokens=4096,
90
+ )
91
+ result = response.choices[0].message.content
92
+ return {"result": result}
93
+ except Exception as e:
94
+ return {"error": f"AI analysis failed: {str(e)}"}
95
 
96
 
97
  class ScanRequest(BaseModel):