luanpoppe commited on
Commit
68d3cc8
·
1 Parent(s): c2e0b1d
endpoint_teste/views.py CHANGED
@@ -3,7 +3,7 @@ from rest_framework import viewsets, filters
3
  from django_filters.rest_framework import DjangoFilterBackend
4
  from endpoint_teste.models import EndpointTesteModel
5
  from endpoint_teste.serializer import EndpointTesteSerializer, PDFUploadSerializer
6
- from ..setup.environment import default_model
7
 
8
  from rest_framework.decorators import api_view
9
  from rest_framework.response import Response
 
3
  from django_filters.rest_framework import DjangoFilterBackend
4
  from endpoint_teste.models import EndpointTesteModel
5
  from endpoint_teste.serializer import EndpointTesteSerializer, PDFUploadSerializer
6
+ from setup.environment import default_model
7
 
8
  from rest_framework.decorators import api_view
9
  from rest_framework.response import Response
langchain_backend/utils.py CHANGED
@@ -7,7 +7,7 @@ from langchain_text_splitters import RecursiveCharacterTextSplitter
7
  from langchain.chains.combine_documents import create_stuff_documents_chain
8
  from langchain_core.prompts import ChatPromptTemplate
9
  from langchain_huggingface import HuggingFaceEndpoint
10
- from ..setup.environment import default_model
11
 
12
  os.environ.get("OPENAI_API_KEY")
13
  os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
7
  from langchain.chains.combine_documents import create_stuff_documents_chain
8
  from langchain_core.prompts import ChatPromptTemplate
9
  from langchain_huggingface import HuggingFaceEndpoint
10
+ from setup.environment import default_model
11
 
12
  os.environ.get("OPENAI_API_KEY")
13
  os.environ.get("HUGGINGFACEHUB_API_TOKEN")