Spaces:
Running
Running
luanpoppe
commited on
Commit
Β·
408e821
1
Parent(s):
d06bd06
fix do Model + refactor pasta de utils
Browse files- {langchain_backend β _utils}/__init__.py +0 -0
- {langchain_backend β _utils}/main.py +2 -2
- {langchain_backend β _utils}/utils.py +0 -0
- endpoint_teste/views.py +1 -1
- modelos_usuarios/migrations/0002_modelousuariomodel_tipo_do_modelo_and_more.py +23 -0
- modelos_usuarios/models.py +1 -1
- resumos/views.py +2 -2
{langchain_backend β _utils}/__init__.py
RENAMED
File without changes
|
{langchain_backend β _utils}/main.py
RENAMED
@@ -1,6 +1,6 @@
|
|
1 |
import os
|
2 |
-
from
|
3 |
-
from
|
4 |
from langchain.chains import create_retrieval_chain
|
5 |
from langchain_huggingface import HuggingFaceEmbeddings
|
6 |
from langchain_chroma import Chroma
|
|
|
1 |
import os
|
2 |
+
from _utils.utils import create_prompt_llm_chain, create_retriever, getPDF, create_llm, create_prompt_llm_chain_summary, process_embedding_summary
|
3 |
+
from _utils import utils
|
4 |
from langchain.chains import create_retrieval_chain
|
5 |
from langchain_huggingface import HuggingFaceEmbeddings
|
6 |
from langchain_chroma import Chroma
|
{langchain_backend β _utils}/utils.py
RENAMED
File without changes
|
endpoint_teste/views.py
CHANGED
@@ -10,7 +10,7 @@ from rest_framework.decorators import api_view, parser_classes
|
|
10 |
from rest_framework.parsers import MultiPartParser
|
11 |
from rest_framework.response import Response
|
12 |
|
13 |
-
from
|
14 |
from .serializer import TesteSerializer
|
15 |
from langchain_huggingface import HuggingFaceEndpoint
|
16 |
|
|
|
10 |
from rest_framework.parsers import MultiPartParser
|
11 |
from rest_framework.response import Response
|
12 |
|
13 |
+
from _utils.main import get_llm_answer
|
14 |
from .serializer import TesteSerializer
|
15 |
from langchain_huggingface import HuggingFaceEndpoint
|
16 |
|
modelos_usuarios/migrations/0002_modelousuariomodel_tipo_do_modelo_and_more.py
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Generated by Django 4.1 on 2024-11-16 00:37
|
2 |
+
|
3 |
+
from django.db import migrations, models
|
4 |
+
|
5 |
+
|
6 |
+
class Migration(migrations.Migration):
|
7 |
+
|
8 |
+
dependencies = [
|
9 |
+
('modelos_usuarios', '0001_initial'),
|
10 |
+
]
|
11 |
+
|
12 |
+
operations = [
|
13 |
+
migrations.AddField(
|
14 |
+
model_name='modelousuariomodel',
|
15 |
+
name='tipo_do_modelo',
|
16 |
+
field=models.CharField(blank=True, max_length=400, null=True),
|
17 |
+
),
|
18 |
+
migrations.AlterField(
|
19 |
+
model_name='modelousuariomodel',
|
20 |
+
name='user_id',
|
21 |
+
field=models.IntegerField(null=True),
|
22 |
+
),
|
23 |
+
]
|
modelos_usuarios/models.py
CHANGED
@@ -3,4 +3,4 @@ from django.db import models
|
|
3 |
class ModeloUsuarioModel(models.Model):
|
4 |
user_id = models.IntegerField(null=True)
|
5 |
modelo = models.TextField(null=False, blank=False)
|
6 |
-
tipo_do_modelo = models.CharField(null=True, blank=True)
|
|
|
3 |
class ModeloUsuarioModel(models.Model):
|
4 |
user_id = models.IntegerField(null=True)
|
5 |
modelo = models.TextField(null=False, blank=False)
|
6 |
+
tipo_do_modelo = models.CharField(null=True, blank=True, max_length=400)
|
resumos/views.py
CHANGED
@@ -2,9 +2,9 @@ from rest_framework.views import APIView
|
|
2 |
import tempfile, os
|
3 |
from rest_framework.response import Response
|
4 |
|
5 |
-
from
|
6 |
from .serializer import ResumoPDFSerializer
|
7 |
-
from
|
8 |
from setup.environment import default_model
|
9 |
from rest_framework.parsers import MultiPartParser
|
10 |
from drf_spectacular.utils import extend_schema
|
|
|
2 |
import tempfile, os
|
3 |
from rest_framework.response import Response
|
4 |
|
5 |
+
from _utils.utils import DEFAULT_SYSTEM_PROMPT
|
6 |
from .serializer import ResumoPDFSerializer
|
7 |
+
from _utils.main import get_llm_answer_summary, get_llm_answer_summary_with_embedding
|
8 |
from setup.environment import default_model
|
9 |
from rest_framework.parsers import MultiPartParser
|
10 |
from drf_spectacular.utils import extend_schema
|