Commit 04df775 1 parent 7722fe4 commit 04df775 Copy full SHA for 04df775
File tree 1 file changed +24
-7
lines changed
1 file changed +24
-7
lines changed Original file line number Diff line number Diff line change 1
- use crate :: models:: general:: llm:: { Message } ;
1
+ use crate :: models:: general:: llm:: Message ;
2
2
3
3
use dotenv:: dotenv;
4
+ use reqwest:: header:: { HeaderMap , HeaderValue } ;
4
5
use std:: env;
6
+ // backend_hive
5
7
6
- use reqwest:: Client ;
7
-
8
-
9
- //call LLM GPT
8
+ // Call LLM GPT
10
9
pub async fn call_gpt ( message : Vec < Message > ) {
11
10
dotenv ( ) . ok ( ) ;
12
11
13
- //extract api keys
14
- let api_keys: String = env:: var ( "GPT_API_KEY" ) . expect ( "GPT_API_KEY" ) ;
12
+ // Extract api keys
13
+ let api_key: String = env:: var ( "OPEN_AI_KEY" ) . expect ( "OPEN_AI_KEY not found" ) ;
14
+ let api_org: String = env:: var ( "OPEN_AI_ORG" ) . expect ( "OPEN_AI_ORG not found" ) ;
15
+
16
+ // Confirm endpoint
17
+ let url: & str = "https://api.openai.com/v1/chat/completions" ;
18
+
19
+ // Create headers
20
+ let mut headers = HeaderMap :: new ( ) ;
21
+
22
+ // Create api key headers
23
+ headers. insert (
24
+ "authorization" ,
25
+ HeaderValue :: from_str ( & format ! ( "Bearer {}" , api_key) ) . unwrap ( ) ,
26
+ ) ;
15
27
28
+ // Create OpenAI org Header
29
+ headers. insert (
30
+ "OpenAI-Organization" ,
31
+ HeaderValue :: from_str ( & api_org. as_str ( ) ) . unwrap ( ) ,
32
+ ) ;
16
33
}
You can’t perform that action at this time.
0 commit comments