pgleeson commited on
Commit
26c8ca0
·
1 Parent(s): d266060

Use temperature for gpt4

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,7 +7,7 @@ from langchain.chains import LLMChain
7
 
8
  import os
9
 
10
- __version__ = '0.1.1'
11
 
12
  LLM_GPT35 = 'GPT3.5'
13
  LLM_GPT4 = 'GPT4'
@@ -38,7 +38,6 @@ def get_openai_api_key():
38
  def get_llamaapi_key():
39
 
40
  llamaapi_key = os.environ.get('LLAMAAPI_KEY')
41
- print('key: [%s]'%llamaapi_key)
42
 
43
  return llamaapi_key
44
 
@@ -58,7 +57,7 @@ def generate_response(input_text, llm_ver, temperature):
58
 
59
  elif llm_ver==LLM_GPT4:
60
  from langchain_openai import ChatOpenAI
61
- llm = ChatOpenAI(model_name='gpt-4', openai_api_key=get_openai_api_key())
62
 
63
  elif llm_ver==LLM_LLAMA2:
64
  from llamaapi import LlamaAPI
 
7
 
8
  import os
9
 
10
+ __version__ = '0.1.2'
11
 
12
  LLM_GPT35 = 'GPT3.5'
13
  LLM_GPT4 = 'GPT4'
 
38
  def get_llamaapi_key():
39
 
40
  llamaapi_key = os.environ.get('LLAMAAPI_KEY')
 
41
 
42
  return llamaapi_key
43
 
 
57
 
58
  elif llm_ver==LLM_GPT4:
59
  from langchain_openai import ChatOpenAI
60
+ llm = ChatOpenAI(model_name='gpt-4', openai_api_key=get_openai_api_key(), temperature=temperature)
61
 
62
  elif llm_ver==LLM_LLAMA2:
63
  from llamaapi import LlamaAPI