luanpoppe commited on
Commit
408e821
Β·
1 Parent(s): d06bd06

fix do Model + refactor pasta de utils

Browse files
{langchain_backend β†’ _utils}/__init__.py RENAMED
File without changes
{langchain_backend β†’ _utils}/main.py RENAMED
@@ -1,6 +1,6 @@
1
  import os
2
- from langchain_backend.utils import create_prompt_llm_chain, create_retriever, getPDF, create_llm, create_prompt_llm_chain_summary, process_embedding_summary
3
- from langchain_backend import utils
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 langchain_backend.main import get_llm_answer
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 langchain_backend.utils import DEFAULT_SYSTEM_PROMPT
6
  from .serializer import ResumoPDFSerializer
7
- from langchain_backend.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
 
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