Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,17 @@
|
|
1 |
import gradio as gr
|
2 |
import os
|
|
|
|
|
|
|
|
|
3 |
|
4 |
GOOGLE_API_KEY = "AIzaSyBbruzn10nez-0a-_60TA9R9h6qumLD1Es" # add your GOOGLE API key here
|
5 |
os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
|
6 |
|
|
|
|
|
|
|
|
|
7 |
|
8 |
from llama_index import (
|
9 |
ServiceContext,
|
@@ -14,11 +22,21 @@ from llama_index import (
|
|
14 |
)
|
15 |
from llama_index.llms import Gemini
|
16 |
from llama_index.embeddings import GeminiEmbedding
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
model_name = "models/embedding-001"
|
19 |
|
20 |
|
21 |
-
llm = Gemini()
|
22 |
embed_model = GeminiEmbedding(
|
23 |
model_name=model_name, api_key=GOOGLE_API_KEY, title="this is a document"
|
24 |
)
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
+
import logging
|
4 |
+
|
5 |
+
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
6 |
+
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
|
7 |
|
8 |
GOOGLE_API_KEY = "AIzaSyBbruzn10nez-0a-_60TA9R9h6qumLD1Es" # add your GOOGLE API key here
|
9 |
os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
|
10 |
|
11 |
+
from g4f import Provider, models
|
12 |
+
from langchain.llms.base import LLM
|
13 |
+
from llama_index.llms import LangChainLLM
|
14 |
+
from langchain_g4f import G4FLLM
|
15 |
|
16 |
from llama_index import (
|
17 |
ServiceContext,
|
|
|
22 |
)
|
23 |
from llama_index.llms import Gemini
|
24 |
from llama_index.embeddings import GeminiEmbedding
|
25 |
+
import g4f
|
26 |
+
g4f.debug.logging = True
|
27 |
+
|
28 |
+
|
29 |
+
llm= LLM = G4FLLM(
|
30 |
+
model=models.gpt_35_turbo_16k,
|
31 |
+
,
|
32 |
+
)
|
33 |
+
|
34 |
+
llm = LangChainLLM(llm=llm)
|
35 |
|
36 |
model_name = "models/embedding-001"
|
37 |
|
38 |
|
39 |
+
#llm = Gemini()
|
40 |
embed_model = GeminiEmbedding(
|
41 |
model_name=model_name, api_key=GOOGLE_API_KEY, title="this is a document"
|
42 |
)
|