Spaces:
Sleeping
Sleeping
Use temperature for gpt4
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from langchain.chains import LLMChain
|
|
| 7 |
|
| 8 |
import os
|
| 9 |
|
| 10 |
-
__version__ = '0.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
|