SuriRaja commited on
Commit
cd403e4
·
verified ·
1 Parent(s): 091b1fd

Delete HuggingFaceAPI.cls

Browse files
Files changed (1) hide show
  1. HuggingFaceAPI.cls +0 -40
HuggingFaceAPI.cls DELETED
@@ -1,40 +0,0 @@
1
- public class HuggingFaceAPI {
2
- private static final String API_URL = 'http://your-flask-api-url/query'; // Change to your Flask API URL
3
-
4
- // Method to make the HTTP request to Hugging Face API
5
- public static String queryHuggingFace(String query) {
6
- HttpRequest req = new HttpRequest();
7
- req.setEndpoint(API_URL);
8
- req.setMethod('POST');
9
- req.setHeader('Content-Type', 'application/json');
10
-
11
- // Prepare the JSON body
12
- String jsonBody = '{"query": "' + query + '"}';
13
- req.setBody(jsonBody);
14
-
15
- Http http = new Http();
16
- HttpResponse res = http.send(req);
17
-
18
- // Check if the response is successful
19
- if (res.getStatusCode() == 200) {
20
- // Parse the JSON response
21
- Map<String, Object> responseMap = (Map<String, Object>)JSON.deserializeUntyped(res.getBody());
22
- String answer = (String)responseMap.get('answer');
23
- return answer;
24
- } else {
25
- System.debug('Error: ' + res.getStatusCode() + ' ' + res.getStatus());
26
- return 'Error: Could not fetch answer from Hugging Face model.';
27
- }
28
- }
29
-
30
- // Method to log query in Salesforce
31
- public static void logQuery(String userId, String query, String answer, String documentLink, Decimal confidenceScore) {
32
- Chat_Query_Log__c newLog = new Chat_Query_Log__c();
33
- newLog.User__c = userId;
34
- newLog.Query__c = query;
35
- newLog.Answer__c = answer;
36
- newLog.Confidence_Score__c = confidenceScore;
37
- newLog.Document_Link__c = documentLink;
38
- insert newLog;
39
- }
40
- }