litellm

1
import React, { useState, useEffect } from 'react';
2

3
const CodeBlock = ({ token }) => {
4
  const codeWithToken = `
5
import os
6
from litellm import completion
7

8
# set ENV variables 
9
os.environ["LITELLM_TOKEN"] = '${token}'
10

11
messages = [{ "content": "Hello, how are you?","role": "user"}]
12

13
# openai call
14
response = completion(model="gpt-3.5-turbo", messages=messages)
15

16
# cohere call
17
response = completion("command-nightly", messages)
18
`;
19

20
  const codeWithoutToken = `
21
from litellm import completion
22

23
## set ENV variables
24
os.environ["OPENAI_API_KEY"] = "openai key"
25
os.environ["COHERE_API_KEY"] = "cohere key"
26

27

28
messages = [{ "content": "Hello, how are you?","role": "user"}]
29

30
# openai call
31
response = completion(model="gpt-3.5-turbo", messages=messages)
32

33
# cohere call
34
response = completion("command-nightly", messages)
35
`;
36
  return (
37
    <pre>
38
        {console.log("token: ", token)}
39
      {token ? codeWithToken : codeWithoutToken}
40
    </pre>
41
  )
42
}
43

44
const QueryParamReader = () => {
45
  const [token, setToken] = useState(null);
46

47
  useEffect(() => {
48
    const urlParams = new URLSearchParams(window.location.search);
49
    console.log("urlParams: ", urlParams)
50
    const token = urlParams.get('token');
51
    setToken(token);
52
  }, []);
53

54
  return (
55
    <div>
56
      <CodeBlock token={token} />
57
    </div>
58
  );
59
}
60

61
export default QueryParamReader;